1 repo
Computational layers for partitioning workloads across processors.
Distinguishing note: Focuses on the underlying engine, distinct from high-level framework features.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Parallel Computing Engines. Refine with filters or upvote what's useful.
ColossalAI is a distributed deep learning framework designed for training and deploying massive artificial intelligence models across clusters of hardware accelerators. It functions as a parallel computing engine that partitions model workloads and data across multiple processors to maximize memory efficiency and throughput. The platform distinguishes itself through a comprehensive suite of parallelization strategies, including multi-dimensional tensor parallelism and pipeline-based model parallelism, which segment neural network layers and stages across devices. To support large-scale genera
Partitions large model workloads and data across multiple processors to maximize memory efficiency and throughput.