Hacker News new | ask | show | jobs
Showh HN: Microjax – JAX in two classes and six functions (github.com)
46 points by joelburget 340 days ago
1 comments
gkbrk 340 days ago
JAX also has a tutorial on re-implementing your own JAX.

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

link