This area is to discuss how to best use JAX on NVIDIA GPUs and discuss problems and issues should they arise.
JAX is a library for high-performance numerical computing and machine learning research.
To help developers to get up and running quickly with JAX, we are working towards a container that includes JAX, FLAX (neural network library), and a set of dependencies tested for performance. Apply for the early access container here.
You can also join our new Discord server to chat within the community using JAX on NVIDIA GPUs
Refer this guide to when reporting a bug or issue