1 repo
Awesome GitHub RepositoriesUniversal Checkpointing
Frameworks that provide hardware-agnostic and topology-independent state persistence for machine learning models.
Distinguishing note: Focuses on the universality of the checkpoint format across varying scales and hardware, distinct from standard serialization.
Explore 1 awesome GitHub repository matching artificial intelligence & ml · Universal Checkpointing. Refine with filters or upvote what's useful.
Awesome Universal Checkpointing GitHub Repositories
Find the best repos with AI.We'll search the best matching repositories with AI.
- deepspeedai/DeepSpeed
deepspeedai/DeepSpeed
41,638View on GitHubDeepSpeed is a high-performance library designed to scale deep learning model training and inference across massive clusters of GPUs and compute nodes. It provides a comprehensive suite of tools for distributed training, enabling the execution of models that exceed the memory capacity of single devices through advanced parameter partitioning, pipeline-based model parallelism, and memory-efficient state offloading. The framework distinguishes itself through specialized communication-efficient optimizers and hardware-aware acceleration techniques. By utilizing gradient compression, quantization
The framework standardizes model, optimizer, and scheduler states into a unified format to enable consistent checkpointing across varying model sizes, topologies, and hardware.
Pythonbillion-parameterscompressiondata-parallelism
