Update jax-ml/jax to commit d79c1c43fe8c40c3c51743e1796f2d2b43ebfb82#1611

wsmoses merged 11 commits intomainEnzymeAD/Enzyme-JAX:mainfrom

update-jax-ml-jaxEnzymeAD/Enzyme-JAX:update-jax-ml-jaxCopy head branch name to clipboard