🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
-
Updated
Nov 18, 2024 - Python
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Deep Learning for humans
Convert Machine Learning Code Between Frameworks
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Trax — Deep Learning with Clear Code and Speed
A library for scientific machine learning and physics-informed learning
Distributed ML Training and Fine-Tuning on Kubernetes
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
Flax is a neural network library for JAX that is designed for flexibility.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Notes, examples, and Python demos for the 2nd edition of the textbook "Machine Learning Refined" (published by Cambridge University Press).
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
A deep learning package for many-body potential energy representation and molecular dynamics
Automated Machine Learning on Kubernetes
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."