🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
-
Updated
May 15, 2024 - Python
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
AI/ML projects to learn and improve
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Automatic Differentiation + Adjoint + Shocks Experiments
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Tevatron - A flexible toolkit for neural retrieval research and development.
Boax is a Bayesian Optimization library for JAX.
Deep Learning for humans
Code I used for my YouTube videos
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.
Orbax provides common utility libraries for JAX users.
Flax is a neural network library for JAX that is designed for flexibility.
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
PEP 503 repository index for jax[cuda]
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."