PrimeIntellect-ai / diloco_simpleLinks
torch implementation of diloco
☆18Updated last year
Alternatives and similar repositories for diloco_simple
Users that are interested in diloco_simple are comparing it to the libraries listed below
Sorting:
- PCCL (Prime Collective Communications Library) implements fault tolerant collective communications over IP☆96Updated this week
- A 7B parameter model for mathematical reasoning☆40Updated 5 months ago
- DeMo: Decoupled Momentum Optimization☆189Updated 7 months ago
- train with kittens!☆61Updated 8 months ago
- Modded vLLM to run pipeline parallelism over public networks☆36Updated 2 months ago
- 👷 Build compute kernels☆77Updated this week
- Simple high-throughput inference library☆120Updated 2 months ago
- Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…☆138Updated 11 months ago
- ☆45Updated last year
- an open source reproduction of NVIDIA's nGPT (Normalized Transformer with Representation Learning on the Hypersphere)☆101Updated 4 months ago
- ☆91Updated this week
- Make triton easier☆47Updated last year
- Official code for "SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient"☆141Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated 4 months ago
- ☆18Updated this week
- Grokking on modular arithmetic in less than 150 epochs in MLX☆14Updated 8 months ago
- ☆49Updated last year
- ☆13Updated last year
- ☆81Updated last year
- NanoGPT (124M) quality in 2.67B tokens☆28Updated 3 weeks ago
- ☆54Updated last year
- Docker image NVIDIA GH200 machines - optimized for vllm serving and hf trainer finetuning☆47Updated 4 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆128Updated 7 months ago
- A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.☆128Updated 3 months ago
- Explore training for quantized models☆20Updated last week
- A MAD laboratory to improve AI architecture designs 🧪☆123Updated 7 months ago
- ☆61Updated last year
- Custom triton kernels for training Karpathy's nanoGPT.☆19Updated 9 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆244Updated 5 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆237Updated last month