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 in PyPI. It can be installed with
pip install ipcmagic[dask]This installs all ipcmagic dependencies including Dask (dask[complete]).
If the Dask support is not needed, ipcmagic can be installed with
pip install ipcmagicwhich doesn't include the installation of Dask and it's dependencies.
See examples here.