Courses
Tutorials
- The PyTorch developer’s guide to JAX fundamentals https://cloud.google.com/blog/products/ai-machine-learning/guide-to-jax-for-pytorch-developers
- Learning JAX as a PyTorch developer https://kidger.site/thoughts/torch2jax/
Exploratory
- The State of the JAX Ecosystem in 2025 https://www.youtube.com/watch?v=TzULI2PomIw