Some basic slurm examples for JAX Sumner cluster - View it on GitHub