Showh HN: Microjax – JAX in two classes and six functions (github.com) 46 points by joelburget 1y ago ↗ HN
[–] gkbrk 1y ago ↗ JAX also has a tutorial on re-implementing your own JAX.https://docs.jax.dev/en/latest/autodidax.html
1 comment
[ 4.7 ms ] story [ 15.2 ms ] threadhttps://docs.jax.dev/en/latest/autodidax.html