blob: ebbcc2fbc0d783d3e4b3009e09222737431a5f5e (
plain)
1
2
3
4
5
6
7
8
9
|
# 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
|