Here are
929 public repositories
matching this topic...
?? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
Updated
May 15, 2024
-
Python
-
Updated
May 15, 2024
-
Python
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
-
Updated
May 15, 2024
-
Python
?? Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
-
Updated
May 15, 2024
-
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
May 12, 2024
-
Python
?? A ranked list of awesome machine learning Python libraries. Updated weekly.
-
Updated
May 14, 2024
-
Python
?? A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
-
Updated
May 15, 2024
-
Python
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
-
Updated
May 12, 2024
-
Python
Trax ? Deep Learning with Clear Code and Speed
-
Updated
Feb 1, 2024
-
Python
Python code for "Probabilistic Machine learning" book by Kevin Murphy
-
Updated
Dec 19, 2023
-
Jupyter Notebook
Flax is a neural network library for JAX that is designed for flexibility.
-
Updated
May 15, 2024
-
Python
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.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
-
Updated
May 15, 2024
-
Python
JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.
-
Updated
Apr 3, 2024
-
Jupyter Notebook
Scenic: A Jax Library for Computer Vision Research and Beyond
-
Updated
May 14, 2024
-
Python
Training and serving large-scale neural networks with auto parallelization.
-
Updated
Dec 9, 2023
-
Python
JAX-based neural network library
-
Updated
May 15, 2024
-
Python
?? A refreshing functional take on deep learning, compatible with your favorite libraries
-
Updated
May 14, 2024
-
Python
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
-
Updated
Apr 3, 2024
-
Python
Improve this page
Add a description, image, and links to the
jax
topic page so that developers can more easily learn about it.
Curate this topic
Add this topic to your repo
To associate your repository with the
jax
topic, visit your repo's landing page and select "manage topics."
Learn more
You can’t perform that action at this time.