Experiments with running JAX on multiple nodes in a slurm environment.
Primary LanguagePython
No issues in this repository yet.