Examples of training models with hybrid parallelism using ColossalAI
☆339Mar 23, 2023Updated 3 years ago
Alternatives and similar repositories for ColossalAI-Examples
Users that are interested in ColossalAI-Examples are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 4 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
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Large-scale model inference.☆629Sep 12, 2023Updated 3 years ago
- ☆30Sep 4, 2023Updated 3 years ago
- Documentation for Colossal-AI☆24Jun 6, 2025Updated last year
- Optimizing AlphaFold Training and Inference on GPU Clusters☆616Jul 16, 2024Updated 2 years ago
- Making large AI models cheaper, faster and more accessible☆41,431Updated this week
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- ☆24Nov 22, 2022Updated 3 years ago
- Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation☆20Jun 11, 2025Updated last year
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆28Jul 11, 2021Updated 5 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 5 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,448Mar 20, 2024Updated 2 years ago
- A curated list of awesome projects and papers for distributed training or inference☆281Oct 8, 2024Updated 2 years ago
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…☆157Apr 10, 2026Updated 6 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Nov 19, 2024Updated last year
- A Controllable Model of Grounded Response Generation (AAAI 21)☆13Oct 25, 2022Updated 4 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,268Aug 14, 2025Updated last year
- ☆462Jun 9, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆36Feb 18, 2026Updated 8 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,756Jan 8, 2024Updated 2 years ago
- Mengzi Pretrained Models☆544Nov 29, 2022Updated 3 years ago
- Example models using DeepSpeed☆6,848Sep 30, 2026Updated last month
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆481Mar 15, 2024Updated 2 years ago
- Ongoing research training transformer models at scale☆18,293Updated this week
- Momentum Decoding: Open-ended Text Generation as Graph Exploration☆19Jan 27, 2023Updated 3 years ago
- generative language model training on top of the JAX and Huggingface 🤗☆10May 11, 2022Updated 4 years ago
- [NeurIPS'22 Spotlight] A Contrastive Framework for Neural Text Generation☆476Mar 7, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆80May 4, 2021Updated 5 years ago
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆318Mar 20, 2023Updated 3 years ago
- OSLO: Open Source for Large-scale Optimization☆178Sep 9, 2023Updated 3 years ago
- A modular RL library to fine-tune language models to human preferences☆2,397Mar 1, 2024Updated 2 years ago
- Transformer related optimization, including BERT, GPT☆6,458Mar 27, 2024Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated 2 years ago
- Training and serving large-scale neural networks with auto parallelization.☆3,176Dec 9, 2023Updated 2 years ago