NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
-
Updated
Jun 9, 2024 - Python
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
🤗 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.
Code I used for my YouTube videos
This is a JAX/Flax-based transformer language model trained on a Japanese dataset. It is based on the official Flax example code (lm1b).
Flax Engine – multi-platform 3D game engine
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. 🚀
Deep Learning examples using the Jax ecosystem of libraries
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
A Library for Uncertainty Quantification.
Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.
A cross-platform launcher for the Flax game engine.
Yet another RL library built with FLAX modules
Tevatron - A flexible toolkit for neural retrieval research and development.
Clean single-file implementation of offline RL algorithms in JAX
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
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."