Check all environment variables for Slurm environment#32799

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom

Flamefire:patch-1Flamefire/jax:patch-1Copy head branch name to clipboard