Jupyter notebook magic commands for running MPI-based Python code and multi-node Dask workloads directly from notebooks.
It supports mpirun and srun (Slurm) launchers, as well as a non-MPI launcher for workloads that do not require inter-rank communication.
This package is called ipcmagic-cscs in PyPI. It can be installed with
pip install ipcmagic-cscs[dask]This installs all ipcmagic-cscs dependencies including Dask (dask[complete]).
If the Dask support is not needed, ipcmagic-cscs can be installed with
pip install ipcmagic-cscswhich doesn't include the installation of Dask and it's dependencies.
See examples here.