3 ms·
JAX 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
by gkbrk 1y ago
JAX 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