Skip to content

Conversation

hanzhi713
Copy link
Member

@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

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