hpcaitech / PaLM-colossalaiLinks
Scalable PaLM implementation of PyTorch
☆190Updated 2 years ago
Alternatives and similar repositories for PaLM-colossalai
Users that are interested in PaLM-colossalai are comparing it to the libraries listed below
Sorting:
- Performance benchmarking with ColossalAI☆39Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆210Updated 10 months ago
- Examples of training models with hybrid parallelism using ColossalAI☆340Updated 2 years ago
- GPTQ inference Triton kernel☆302Updated 2 years ago
- Running BERT without Padding☆471Updated 3 years ago
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆208Updated last year
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆221Updated last year
- Fast Inference Solutions for BLOOM☆564Updated 8 months ago
- ☆119Updated last year
- A unified tokenization tool for Images, Chinese and English.☆152Updated 2 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- ☆105Updated 2 years ago
- ☆96Updated 2 years ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆96Updated last year
- Large-scale model inference.☆629Updated last year
- A collection of models built with ColossalAI☆32Updated 2 years ago
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆313Updated 2 years ago
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆57Updated last year
- 📑 Dive into Big Model Training☆114Updated 2 years ago
- Microsoft Automatic Mixed Precision Library☆609Updated 8 months ago
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆295Updated 3 months ago
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆95Updated last year
- This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/…☆95Updated last year
- DSIR large-scale data selection framework for language model training☆251Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆116Updated 6 months ago
- ☆147Updated 2 years ago
- Inference script for Meta's LLaMA models using Hugging Face wrapper☆110Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆268Updated 3 years ago
- ☆79Updated last year
- Model Compression for Big Models☆162Updated last year