TensorFlow works with Slurm Interactive Session but Not Slurm Job

Question

I\'m trying to get some TensorFlow/Jax code to run on a Slurm cluster\'s GPUs. When I request an interactive GPU session and run my code, everything works perfectly. But whe