Resources
Here is a list of resources to get started with JAX.
Official documentation
Other resources
Awesome JAX is a great list of resources on JAX.
Q&A
Alliance wiki
There is currently no page on JAX, but there is a page on Flax.
Marie-Hélène Burle
Here is a list of resources to get started with JAX.
Awesome JAX is a great list of resources on JAX.
There is currently no page on JAX, but there is a page on Flax.