hpcaitech / PaLM-colossalaiLinks
Scalable PaLM implementation of PyTorch
☆189Updated 3 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:
- ☆105Updated 2 years ago
- Open Instruction Generalist is an assistant trained on massive synthetic instructions to perform many millions of tasks☆210Updated last year
- Large Scale Distributed Model Training strategy with Colossal AI and Lightning AI☆56Updated 2 years ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆225Updated 2 years ago
- ☆98Updated 2 years ago
- Fast Inference Solutions for BLOOM☆565Updated last year
- Simple implementation of Speculative Sampling in NumPy for GPT-2.☆98Updated 2 years ago
- 📑 Dive into Big Model Training☆116Updated 3 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆339Updated 2 years ago
- Train llm (bloom, llama, baichuan2-7b, chatglm3-6b) with deepspeed pipeline mode. Faster than zero/zero++/fsdp.☆98Updated last year
- ☆122Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Updated 2 years ago
- [ACL 2022] Structured Pruning Learns Compact and Accurate Models https://arxiv.org/abs/2204.00408☆198Updated 2 years ago
- ☆24Updated 3 years ago
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆317Updated 2 years ago
- A unified tokenization tool for Images, Chinese and English.☆153Updated 2 years ago
- Techniques used to run BLOOM at inference in parallel☆37Updated 3 years ago
- Inference script for Meta's LLaMA models using Hugging Face wrapper☆110Updated 2 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆113Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆219Updated last year
- A collection of models built with ColossalAI☆32Updated 3 years ago
- The aim of this repository is to utilize LLaMA to reproduce and enhance the Stanford Alpaca☆98Updated 2 years ago
- This PyTorch package implements MoEBERT: from BERT to Mixture-of-Experts via Importance-Guided Adaptation (NAACL 2022).☆112Updated 3 years ago
- Princeton NLP's pre-training library based on fairseq with DeepSpeed kernel integration 🚃☆114Updated 3 years ago
- A plug-in of Microsoft DeepSpeed to fix the bug of DeepSpeed pipeline☆25Updated 4 years ago
- Repository for analysis and experiments in the BigCode project.☆128Updated last year
- [NeurIPS'23] Speculative Decoding with Big Little Decoder☆95Updated last year
- An experimental implementation of the retrieval-enhanced language model☆75Updated 2 years ago
- DSIR large-scale data selection framework for language model training☆266Updated last year
- A (somewhat) minimal library for finetuning language models with PPO on human feedback.☆88Updated 3 years ago