#!/bin/bash
#SBATCH --time=00:05:00
#SBATCH --nodes=2
#SBATCH --ntasks=3
#SBATCH --job-name=python_mpi
#SBATCH --mem=8000
 
module purge
module load mpi4py/4.1.0-gompi-2025a
 
mpirun python ./python_mpi.py