jax

fix: reduce DPX workers_per_gpu 2→1 to avoid HBM exhaustion

#890

Open