jax
da40f36d - Fix run_multi_gpu script multi-gpu issue and refactor code (#208)

Commit

1 year ago

Fix run_multi_gpu script multi-gpu issue and refactor code (#208)

References

#208 - Fix run_multi_gpu script multi-gpu issue and refactor code

#266 - Update XLA Commit hash

#305 - Add support for ROCm wheel based install

Author

Ruturaj4

Ruturaj4

Parents

Loading