[Python][ML] Add JAX output format to RDataLoader#22550

Merged

siliataider merged 3 commits into

root-project:masterfrom

siliataider:rdataloader

Jun 12, 2026

Commits on Jun 9, 2026

[Python][ML] Refactor batch conversion into helpers
siliataider
committed

Commits on Jun 10, 2026

[Python][ML] Add as_jax() output format to RDataLoader
siliataider
committed

Commits on Jun 12, 2026

[Python][ML] Add test for the jax output format
siliataider
committed