Under Arch Linux you will need the essential software

$ yay -Syu python-py-pde python-findiff python-pystencils # mandatory
$ yay -S ffmpeg python-h5py-openmpi jupyterlab-widgets python-numba-mpi python-pandas python-pyfftw python-rocket-fft # optional

For tutorial we are following this template containerd image.