hpcaitech / PaLM-colossalai
Scalable PaLM implementation of PyTorch
☆192Updated 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
- Performance benchmarking with ColossalAI☆39Updated 2 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆337Updated last year
- ☆105Updated last year
- GPTQ inference Triton kernel☆297Updated last year
- Official repository for LightSeq: Sequence Level Parallelism for Distributed Training of Long Context Transformers☆206Updated 6 months ago
- ☆96Updated last year
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆308Updated last year
- Fast Inference Solutions for BLOOM☆563Updated 4 months ago
- ☆116Updated 11 months ago
- DSIR large-scale data selection framework for language model training☆241Updated 10 months ago
- This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/…☆96Updated 11 months ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆215Updated last year
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆91Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆106Updated 3 months ago
- Running BERT without Padding☆470Updated 2 years ago
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆208Updated last year
- Inference script for Meta's LLaMA models using Hugging Face wrapper☆110Updated last year
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆93Updated last year
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆57Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated last year
- Zero Bubble Pipeline Parallelism☆345Updated 3 weeks ago
- A unified tokenization tool for Images, Chinese and English.☆151Updated last year
- Experiments on speculative sampling with Llama models☆125Updated last year
- REST: Retrieval-Based Speculative Decoding, NAACL 2024☆194Updated 3 months ago
- Large-scale model inference.☆628Updated last year
- A (somewhat) minimal library for finetuning language models with PPO on human feedback.☆86Updated 2 years ago
- ☆76Updated last year
- Transformer related optimization, including BERT, GPT☆59Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆85Updated 4 months ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆193Updated last year