Taking over 1s now on M1 Mac and cluster CPUs. Abundances also seems to have slowed down, from 10s of ms to 200 ms. JAX version issue?