Skip to content
Navigation Menu
Toggle navigation
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
New
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Search or jump to...
Search code, repositories, users, issues, pull requests...
Search syntax tips
Provide feedback
Saved searches
Use saved searches to filter your results more quickly
Sign in
Sign up
Appearance settings
Resetting focus
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
ray-project
/
ray
Public
Notifications
You must be signed in to change notification settings
Fork
7.8k
Star
43.1k
Code
Issues
2.8k
Pull requests
636
Actions
Projects
Security and quality
4
Insights
Additional navigation options
Code
Issues
Pull requests
Actions
Projects
Security and quality
Insights
[data] add jax data iterator
#61630
Merged
aslonnie
merged 16 commits into
ray-project:master
ray-project/ray:master
from
siyuanfoundation:jax-dataloader
siyuanfoundation/ray:jax-dataloader
Copy head branch name to clipboard
Apr 14, 2026
Conversation
Commits
16
(16)
Checks
Files changed
Merged
[data] add jax data iterator
#61630
aslonnie
merged 16 commits into
ray-project:master
ray-project/ray:master
from
siyuanfoundation:jax-dataloader
siyuanfoundation/ray:jax-dataloader
Copy head branch name to clipboard
Commits
Commits on Apr 14, 2026
[data] add jax data iterator
Show description for f6f5c8c
siyuanfoundation
committed
f6f5c8c
Copy full SHA for f6f5c8c
[data] address review comments for iter_jax_batches
Show description for dfebf8f
siyuanfoundation
committed
dfebf8f
Copy full SHA for dfebf8f
Signed-off-by: siyuanfoundation <sizhang@google.com>
Show description for 1f905b4
siyuanfoundation
committed
1f905b4
Copy full SHA for 1f905b4
[data] Add jax util test to BUILD.bazel
Show description for 17dc463
siyuanfoundation
committed
17dc463
Copy full SHA for 17dc463
[data] set synchronize_batches default to false for jax_sync_generator
Show description for 2b8acb5
siyuanfoundation
committed
2b8acb5
Copy full SHA for 2b8acb5
[data] change jax namesharding param to more flexible transform func
Show description for 0907a1b
siyuanfoundation
committed
0907a1b
Copy full SHA for 0907a1b
[data] remove transform func from iter_jax_batches
Show description for c0d28e8
siyuanfoundation
committed
c0d28e8
Copy full SHA for c0d28e8
[data] add collate_fn to iter_jax_batches
Show description for cb209bc
siyuanfoundation
committed
cb209bc
Copy full SHA for cb209bc
[data] fix pyrefly error
Show description for 9649bb0
siyuanfoundation
committed
9649bb0
Copy full SHA for 9649bb0
[data] Upgrade JAX to 0.4.32 and simplify jax_util sharding logic
Show description for dc4ba61
siyuanfoundation
committed
dc4ba61
Copy full SHA for dc4ba61
[data] Add padding support to iter_jax_batches for fixed-size JAX arrays
Show description for 00afbd4
siyuanfoundation
committed
00afbd4
Copy full SHA for 00afbd4
[data] support dtypes in iter_jax_batches
Show description for 61f72ec
siyuanfoundation
committed
61f72ec
Copy full SHA for 61f72ec
[data] make batch_size non-optional for iter_jax_batches
Show description for 7745562
siyuanfoundation
committed
7745562
Copy full SHA for 7745562
[data] add more explicit type hints to jax_sync_generator
Show description for 05157a4
siyuanfoundation
committed
05157a4
Copy full SHA for 05157a4
[data] rename pad_token_ids to paddings
Show description for 3b70602
siyuanfoundation
committed
3b70602
Copy full SHA for 3b70602
[data] fix jax dependency
Show description for a3f9b19
siyuanfoundation
committed
a3f9b19
Copy full SHA for a3f9b19
You can’t perform that action at this time.
