This website requires JavaScript.
Explore
Help
Sign In
mirrors
/
rocm_jax
Watch
1
Star
0
Fork
0
You've already forked rocm_jax
mirror of
https://github.com/ROCm/jax.git
synced
2025-04-23 22:56:04 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
rocm_jax
/
jax
/
_src
/
clusters
History
Yunlong Liu
609dfac294
Adds a flag to control proxy env checking.
...
name typo fix. Fixes comments.
2024-11-16 00:00:59 -08:00
..
__init__.py
Cleanup: fix unused imports & mark exported names
2024-10-16 17:42:41 -07:00
cloud_tpu_cluster.py
Avoid querying metadata query to check if it's GCE if
TPU_SKIP_MDS_QUERY
is set.
2024-10-23 10:09:02 -07:00
cluster.py
Adds a flag to control proxy env checking.
2024-11-16 00:00:59 -08:00
k8s_cluster.py
add k8s cluster environment
2024-09-20 17:26:53 +00:00
mpi4py_cluster.py
Merge pull request
#20174
from coreyjadams:main
2024-07-08 12:19:18 -07:00
ompi_cluster.py
Squashed commit of the following:
2024-07-02 13:18:05 -05:00
slurm_cluster.py
Squashed commit of the following:
2024-07-02 13:18:05 -05:00