Jax
https://moocaholic.medium.com/jax-a13e83f49897
jax
,
autograd