module purge ml Stages/2022 GCC OpenMPI TensorFlow Horovod mpi4py PyTorch scikit-learn Jupyter CUDA OpenCV \ matplotlib PyTorch-Geometric ASE numba Pandoc tensorboardX PyYAML Pillow-SIMD JupyterKernel-PyDeepLearning tqdm