Does Numba support MPI and/or openMP parallelization?

[stuartarchibald]

Hi @goldmosh,

@goldmosh

I want to know if Numba supports MPI parallelization.

Not out of the box. You can use a lot of ctypes in Numba and could call MPI functions if you wanted to but it’d probably be a lot of work. You might be interested in trying out dask and it’s dask.distributed backend, it works well with Numba.

@goldmosh

I also want to know if Numba supports openMP (shared memory) parallelization.

Yes. See Automatic parallelization with @jit — Numba 0+untagged.4124.gd4460fe.dirty documentation, you can even elect to use OpenMP as the threading layer.

Hope this helps?