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

39 joelburget 1 7/7/2025, 1:41:55 PM github.com ↗

Comments (1)

gkbrk · 10h ago

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

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