Scalable Slurm Cluster for GCP with some JAX bits sprinkled in. - View it on GitHub