-
Updated
Feb 8, 2023 - Python
jax
Here are 536 public repositories matching this topic...
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
Updated
Feb 8, 2023 - Python
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 400 universities from 60 countries including Stanford, MIT, Harvard, and Cambridge.
-
Updated
Feb 4, 2023 - Python
-
Updated
Feb 2, 2023
The Unified Machine Learning Framework
-
Updated
Feb 8, 2023 - Python
Trax — Deep Learning with Clear Code and Speed
-
Updated
Dec 19, 2022 - Python
Deep learning operations reinvented (for pytorch, tensorflow, jax and others)
-
Updated
Jan 18, 2023 - Python
Python code for "Probabilistic Machine learning" book by Kevin Murphy
-
Updated
Jan 20, 2023 - 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
Jan 21, 2022
Flax is a neural network library for JAX that is designed for flexibility.
-
Updated
Feb 8, 2023 - Python
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
-
Updated
Feb 8, 2023 - Python
-
Updated
Feb 8, 2023 - Python
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
-
Updated
Nov 22, 2022 - Python
JAX-based neural network library
-
Updated
Feb 6, 2023 - Python
Fast and Easy Infinite Neural Networks in Python
-
Updated
Jan 21, 2023 - Jupyter Notebook
Scenic: A Jax Library for Computer Vision Research and Beyond
-
Updated
Feb 6, 2023 - Python
Training and serving large-scale neural networks
-
Updated
Feb 7, 2023 - Python
PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
-
Updated
Feb 8, 2023 - Python
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
-
Updated
Feb 8, 2023 - Python
Improve this page
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."

