jax
Here are 1,606 public repositories matching this topic...
Language:All
Sort:Most stars
Deep Learning for humans
- Updated
Feb 19, 2026 - Python
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- Updated
Feb 19, 2026 - Python
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
- Updated
Aug 18, 2024 - Python
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
- Updated
Feb 19, 2026
Convert Machine Learning Code Between Frameworks
- Updated
Feb 7, 2026 - Python
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
- Updated
Feb 19, 2026 - Python
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
- Updated
Feb 15, 2026 - Python
Trax — Deep Learning with Clear Code and Speed
- Updated
Sep 26, 2025 - Python
Flax is a neural network library for JAX that is designed for flexibility.
- Updated
Feb 19, 2026 - Jupyter Notebook
Python code for "Probabilistic Machine learning" book by Kevin Murphy
- Updated
Nov 30, 2025 - Jupyter Notebook
It is my belief that you, the postgraduate students and job-seekers for whom the book is primarily meant will benefit from reading it; however, it is my hope that even the most experienced researchers will find it fascinating as well.
- Updated
Aug 22, 2025
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
- Updated
Apr 3, 2024 - Jupyter Notebook
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
- Updated
Feb 18, 2026 - Python
A library for scientific machine learning and physics-informed learning
- Updated
Dec 12, 2025 - Python
Scenic: A Jax Library for Computer Vision Research and Beyond
- Updated
Feb 18, 2026 - Python
JAX-based neural network library
- Updated
Jan 30, 2026 - Python
Training and serving large-scale neural networks with auto parallelization.
- Updated
Dec 9, 2023 - Python
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.
- Updated
Feb 18, 2026 - Python
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
- Updated
Oct 31, 2025 - Jupyter Notebook
Improve this page
Add a description, image, and links to thejax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with thejax topic, visit your repo's landing page and select "manage topics."