Possible cause of hung workflows at the last stage:
/u/aurora-r0/staniewi/miniconda3/envs/mapping-311/lib/python3.11/multiprocessing/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.
self.pid = os.fork()
Possible cause of hung workflows at the last stage: