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

1 comments

• gkbrk 2 days ago

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

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