# Optimization with JAX

Just me learning optimization with JAX.

## References

https://danielrothenberg.com/blog/2020/Sep/jax-first-steps-pt1/
https://jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html
https://sjmielke.com/jax-purify.htm