Skip to content

Conversation

@hanzhi713
Copy link
Contributor

@hanzhi713 hanzhi713 commented Jul 11, 2025

Due to a bug in jax 0.5.3, you need to apply a patch to jax by prepending the following command to training command

patch /opt/venv/lib/python3.10/site-packages/jax/experimental/shard_map.py -p0 < patches/shard_map.py.patch

@github-actions
Copy link

This pull request has been automatically marked as stale because it has been inactive for 60 days. It will be closed in 7 days if no further activity occurs. If you would like to continue working on this, please remove the stale label or leave a comment.

@github-actions github-actions bot added the stale label Oct 19, 2025
@github-actions github-actions bot removed the stale label Oct 30, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant