AI-Hypercomputer / maxdiffusionLinks
☆238Updated this week
Alternatives and similar repositories for maxdiffusion
Users that are interested in maxdiffusion are comparing it to the libraries listed below
Sorting:
- Google TPU optimizations for transformers models☆117Updated 6 months ago
- ☆142Updated 2 weeks ago
- JAX implementation of the Llama 2 model☆219Updated last year
- ☆187Updated this week
- JAX-Toolbox☆327Updated this week
- Scalable and Performant Data Loading☆290Updated last week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆130Updated last year
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆372Updated this week
- Focused on fast experimentation and simplicity☆76Updated 7 months ago
- Load compute kernels from the Hub☆220Updated this week
- Implementation of Flash Attention in Jax☆215Updated last year
- ☆83Updated last year
- ☆64Updated 8 months ago
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆522Updated this week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆66Updated 4 months ago
- ☆304Updated last year
- DeMo: Decoupled Momentum Optimization☆190Updated 8 months ago
- ☆141Updated this week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆364Updated last month
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆35Updated 9 months ago
- supporting pytorch FSDP for optimizers☆84Updated 7 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆149Updated last month
- Modular, scalable library to train ML models☆142Updated this week
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆127Updated 8 months ago
- ☆323Updated last month
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆94Updated last week
- A simple library for scaling up JAX programs☆140Updated 9 months ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆291Updated 11 months ago
- Experiment of using Tangent to autodiff triton☆79Updated last year