hpcaitech / PaLM-colossalai
Scalable PaLM implementation of PyTorch
☆192Updated last year
Related projects ⓘ
Alternatives and complementary repositories for PaLM-colossalai
- Performance benchmarking with ColossalAI☆39Updated 2 years ago
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆196Updated 2 months ago
- ☆109Updated 7 months ago
- GPTQ inference Triton kernel☆283Updated last year
- Examples of training models with hybrid parallelism using ColossalAI☆336Updated last year
- Fast Inference Solutions for BLOOM☆560Updated last month
- Inference script for Meta's LLaMA models using Hugging Face wrapper☆111Updated last year
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆206Updated 9 months ago
- Running BERT without Padding☆460Updated 2 years ago
- Latency and Memory Analysis of Transformer Models for Training and Inference☆352Updated 5 months ago
- ☆102Updated last year
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆253Updated 2 months ago
- 📑 Dive into Big Model Training☆110Updated last year
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆352Updated last week
- ☆94Updated last year
- Techniques used to run BLOOM at inference in parallel☆37Updated 2 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- DSIR large-scale data selection framework for language model training☆227Updated 7 months ago
- Experiments on speculative sampling with Llama models☆117Updated last year
- A collection of models built with ColossalAI☆32Updated last year
- Triton-based implementation of Sparse Mixture of Experts.☆184Updated 3 weeks ago
- Microsoft Automatic Mixed Precision Library☆522Updated last month
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆74Updated 3 weeks ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆89Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆96Updated this week
- ☆198Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆67Updated 3 months ago
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆58Updated last year
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆305Updated last year
- ☆411Updated 11 months ago