This seems to be triggered by use of numba functions. It may be related to import issues. No problem in Ray or MPI in SPMD mode.