Examples of training models with hybrid parallelism using ColossalAI
☆339Mar 23, 2023Updated 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:
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 3 years ago
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- A collection of models built with ColossalAI☆32Nov 22, 2022Updated 3 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Nov 27, 2024Updated last year
- Large-scale model inference.☆627Sep 12, 2023Updated 2 years ago
- ☆30Sep 4, 2023Updated 2 years ago
- GPT Demo with hybrid distributed training☆10Dec 1, 2022Updated 3 years ago
- Making large AI models cheaper, faster and more accessible☆41,362Updated this week
- ☆24Nov 22, 2022Updated 3 years ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional acc…☆38Jun 1, 2021Updated 4 years ago
- ☆28Jul 11, 2021Updated 4 years ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 2 years ago
- A curated list of awesome projects and papers for distributed training or inference☆267Oct 8, 2024Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,437Mar 20, 2024Updated 2 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Nov 19, 2024Updated last year
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…☆152Jun 25, 2025Updated 8 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,233Aug 14, 2025Updated 7 months ago
- a simple demo to implement a deep learning frame based on static graph☆20Apr 7, 2022Updated 3 years ago
- Momentum Decoding: Open-ended Text Generation as Graph Exploration☆19Jan 27, 2023Updated 3 years ago
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆477Mar 15, 2024Updated 2 years ago
- ☆78May 4, 2021Updated 4 years ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,739Jan 8, 2024Updated 2 years ago
- Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation☆19Jun 11, 2025Updated 9 months ago
- Ongoing research training transformer models at scale☆15,647Updated this week
- OSLO: Open Source for Large-scale Optimization☆175Sep 9, 2023Updated 2 years ago
- Example models using DeepSpeed☆6,805Mar 4, 2026Updated 2 weeks ago
- A utility for storing and reading files for Korean LM training 💾☆35Oct 15, 2025Updated 5 months ago
- LLM training technologies developed by kwai☆71Jan 21, 2026Updated last month
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆319Mar 20, 2023Updated 3 years ago
- Mengzi Pretrained Models☆542Nov 29, 2022Updated 3 years ago
- kogpt를 oslo로 파인튜닝하는 예제.☆23Aug 26, 2022Updated 3 years ago
- Transformer related optimization, including BERT, GPT☆6,397Mar 27, 2024Updated last year
- A Controllable Model of Grounded Response Generation (AAAI 21)☆13Oct 25, 2022Updated 3 years ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆147Jun 25, 2022Updated 3 years ago
- Efficient AI Inference & Serving☆480Jan 8, 2024Updated 2 years ago
- ☆461Jun 9, 2024Updated last year
- Accepted to MLSys 2026☆72Mar 5, 2026Updated 2 weeks ago