This seems to be fixed in the latest release (0.0.5)

Can confirm that this is till broken at the latest release (0.0.5)

no reply from Apple? Does someone monitor these forums?

Hi @dingshuhan any update on this release? The bug on scatter means that jacobians in Jax can't be computed on jax-metal.