jax
b8d2dede - [ROCm] Change run_multi_gpu set opts

Commit

1 year ago

[ROCm] Change run_multi_gpu set opts

References

#169 - [JAX] Ci qa sync

#266 - Update XLA Commit hash

#305 - Add support for ROCm wheel based install

Author

Ruturaj4

Ruturaj4

Committer

Ruturaj4

Ruturaj4

Parents

Loading