erfanzar / EasyDeL
Accelerate, Optimize performance with streamlined training and serving options with JAX.
☆208Updated this week
Related projects ⓘ
Alternatives and complementary repositories for EasyDeL
- paralleled/unparalleled computational with FJFormer☆24Updated last week
- Multipack distributed sampler for fast padding-free training of LLMs☆178Updated 3 months ago
- Inference code for LLaMA models in JAX☆113Updated 6 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆194Updated this week
- JAX implementation of the Llama 2 model☆210Updated 9 months ago
- ☆225Updated 4 months ago
- Google TPU optimizations for transformers models☆75Updated this week
- Understand and test language model architectures on synthetic tasks.☆163Updated 6 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆214Updated this week
- ☆73Updated 4 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆518Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆84Updated this week
- seqax = sequence modeling + JAX☆134Updated 4 months ago
- LoRA for arbitrary JAX models and functions☆133Updated 8 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆113Updated 7 months ago
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆87Updated 3 months ago
- code for training & evaluating Contextual Document Embedding models☆119Updated this week
- ☆197Updated 4 months ago
- JAX Synergistic Memory Inspector☆164Updated 4 months ago
- Train very large language models in Jax.☆195Updated last year
- ☆64Updated 2 years ago
- Flash Attention Implementation with Multiple Backend Support and Sharding This module provides a flexible implementation of Flash Attenti…☆18Updated last week
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆80Updated 11 months ago
- some common Huggingface transformers in maximal update parametrization (µP)☆77Updated 2 years ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆477Updated 3 weeks ago
- Manage scalable open LLM inference endpoints in Slurm clusters☆238Updated 4 months ago
- Normalized Transformer (nGPT)☆87Updated this week
- Fast bare-bones BPE for modern tokenizer training☆142Updated last month
- Language models scale reliably with over-training and on downstream tasks☆94Updated 7 months ago
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆262Updated last year