1 repo
Awesome GitHub RepositoriesKernel Optimization Libraries
High-performance computational kernels for accelerating neural network operations.
Distinguishing note: Focuses on custom mathematical kernels, distinct from general model parallelism.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Kernel Optimization Libraries. Refine with filters or upvote what's useful.
Awesome Kernel Optimization Libraries GitHub Repositories
Find the best repos with AI.We'll search the best matching repositories with AI.
- hpcaitech/ColossalAI
hpcaitech/ColossalAI
41,349View on GitHubColossalAI 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
Replaces standard operations with custom high-performance kernels to accelerate mathematical calculations.
Pythonaibig-modeldata-parallelism
