🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
-
Updated
Jun 10, 2024 - Python
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
Deep Learning for humans
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Automated Machine Learning on Kubernetes
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
Orbax provides common utility libraries for JAX users.
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.
Digital-analog quantum programming interface
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Machine learning algorithms for many-body quantum systems
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."