hpcaitech / ColossalAI-ExamplesLinks
Examples of training models with hybrid parallelism using ColossalAI
☆340Updated 2 years ago
Alternatives and similar repositories for ColossalAI-Examples
Users that are interested in ColossalAI-Examples are comparing it to the libraries listed below
Sorting:
- Scalable PaLM implementation of PyTorch☆190Updated 2 years ago
- Fast Inference Solutions for BLOOM☆565Updated 11 months ago
- Large-scale model inference.☆632Updated 2 years ago
- Efficient Inference for Big Models☆589Updated 2 years ago
- LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training☆408Updated last month
- Plain pytorch implementation of LLaMA☆188Updated 2 years ago
- Collaborative Training of Large Language Models in an Efficient Way☆416Updated last year
- Efficient Training (including pre-training and fine-tuning) for Big Models☆609Updated last month
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,418Updated last year
- A unified tokenization tool for Images, Chinese and English.☆151Updated 2 years ago
- [NIPS2023] RRHF & Wombat☆812Updated 2 years ago
- ☆460Updated last year
- Crosslingual Generalization through Multitask Finetuning☆536Updated last year
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆224Updated last year
- Rectified Rotary Position Embeddings☆381Updated last year
- Open-source pre-training implementation of Google's LaMDA in PyTorch. Adding RLHF similar to ChatGPT.☆472Updated last year
- Official repository for LongChat and LongEval☆533Updated last year
- Automatically split your PyTorch models on multiple GPUs for training & inference☆657Updated last year
- GPTQ inference Triton kernel☆308Updated 2 years ago
- The CUDA version of the RWKV language model ( https://github.com/BlinkDL/RWKV-LM )☆223Updated 9 months ago
- A collection of models built with ColossalAI☆32Updated 2 years ago
- SwissArmyTransformer is a flexible and powerful library to develop your own Transformer variants.☆1,089Updated 9 months ago
- Multi-language Enhanced LLaMA☆302Updated 2 years ago
- Model Compression for Big Models☆165Updated 2 years ago
- Performance benchmarking with ColossalAI☆39Updated 3 years ago
- This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/…☆97Updated last year
- ☆412Updated last year
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆314Updated 2 years ago
- LOMO: LOw-Memory Optimization☆989Updated last year
- PatrickStar enables Larger, Faster, Greener Pretrained Models for NLP and democratizes AI for everyone.☆767Updated 2 years ago