/slurm-jax

Experiments with running JAX on multiple nodes in a slurm environment.

Primary LanguagePython

Stargazers