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

no reply from Apple? Does someone monitor these forums?

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

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