Showh HN: Microjax – JAX in two classes and six functions github.com 45 points by joelburget 2 days ago
gkbrk 2 days ago JAX also has a tutorial on re-implementing your own JAX.https://docs.jax.dev/en/latest/autodidax.html
JAX also has a tutorial on re-implementing your own JAX.
https://docs.jax.dev/en/latest/autodidax.html