News for hackers. Stuff that matters.
Re:Showh HN: Microjax – JAX in two classes and six functions by gkbrk on Monday July 07, 2025 @10:02AM (#44490480)
JAX also has a tutorial on re-implementing your own JAX.

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