🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
-
Updated
Jun 30, 2024 - Python
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
Code I used for my YouTube videos
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
Orbax provides common utility libraries for JAX users.
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
Ready to use, open-source framework for creating games in Flax
Dear ImGui plugin for Flax Engine that adds debug GUI interface to game viewport.
Flax Engine – multi-platform 3D game engine
A Simplistic trainer for Flax
Tevatron - A flexible toolkit for neural retrieval research and development.
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Clean single-file implementation of offline RL algorithms in JAX
Repository for "Blending Data-Driven Priors in Dynamic Games" - RSS 2024
Deep Learning examples using the Jax ecosystem of libraries
Implementation of OpenAI's CLIP in JAX/Flax
A Library for Uncertainty Quantification.
Add a description, image, and links to the flax topic page so that developers can more easily learn about it.
To associate your repository with the flax topic, visit your repo's landing page and select "manage topics."