flax
87e7fa75 - Update Scale up Flax Modules on multiple devices

Commit

2 years ago

Update Scale up Flax Modules on multiple devices

References

#3016 - Upgrade the `pjit` guide to `jax.jit`

Author

8bitmp3

8bitmp3

Committer

IvyZX

IvyZX

Parents

Loading