Showh HN: Microjax – JAX in two classes and six functions

github.com/joelburget

46 pointsjoelburget1 year ago1 comment

JAX also has a tutorial on re-implementing your own JAX.

https://docs.jax.dev/en/latest/autodidax.html