🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
-
Updated
Jun 13, 2024 - Python
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Code I used for my YouTube videos
Orbax provides common utility libraries for JAX users.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Flax Engine – multi-platform 3D game engine
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
A cross-platform launcher for the Flax game engine.
This is a JAX/Flax-based transformer language model trained on a Japanese dataset. It is based on the official Flax example code (lm1b).
A JAX-based deep learning framework for image segmentation using diffusion models.
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
Deep Learning examples using the Jax ecosystem of libraries
A Library for Uncertainty Quantification.
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.
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."