abacusai / gh200-llmLinks
Docker image NVIDIA GH200 machines - optimized for vllm serving and hf trainer finetuning
☆46Updated 4 months ago
Alternatives and similar repositories for gh200-llm
Users that are interested in gh200-llm are comparing it to the libraries listed below
Sorting:
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆128Updated 7 months ago
- train with kittens!☆61Updated 8 months ago
- ArcticTraining is a framework designed to simplify and accelerate the post-training process for large language models (LLMs)☆156Updated this week
- Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".☆277Updated last year
- 👷 Build compute kernels☆77Updated this week
- An efficent implementation of the method proposed in "The Era of 1-bit LLMs"☆153Updated 9 months ago
- Load compute kernels from the Hub☆207Updated this week
- ☆199Updated 7 months ago
- The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …☆60Updated 9 months ago
- The simplest implementation of recent Sparse Attention patterns for efficient LLM inference.☆78Updated last month
- Long context evaluation for large language models☆220Updated 4 months ago
- ☆80Updated last year
- Experiments on speculative sampling with Llama models☆128Updated 2 years ago
- Example of applying CUDA graphs to LLaMA-v2☆12Updated last year
- 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
- ☆214Updated 5 months ago
- Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"☆244Updated 5 months ago
- Official code for "SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient"☆141Updated last year
- Experiment of using Tangent to autodiff triton☆79Updated last year
- ring-attention experiments☆143Updated 9 months ago
- PCCL (Prime Collective Communications Library) implements fault tolerant collective communications over IP☆96Updated last month
- vLLM: A high-throughput and memory-efficient inference and serving engine for LLMs☆87Updated this week
- Collection of autoregressive model implementation☆85Updated 2 months ago
- Manage scalable open LLM inference endpoints in Slurm clusters☆265Updated last year
- The evaluation framework for training-free sparse attention in LLMs☆83Updated 3 weeks ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆256Updated last week
- Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)☆190Updated last year
- Make triton easier☆47Updated last year
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆173Updated 6 months ago
- Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*☆85Updated last year