5 ms·Showh HN: Microjax – JAX in two classes and six functions46 points by joelburget 1y agogkbrk 1y agoJAX also has a tutorial on re-implementing your own JAX. https://docs.jax.dev/en/latest/autodidax.html https://docs.jax.dev/en/latest/autodidax.html