colossal-ai topic

List colossal-ai repositories

CachedEmbedding

91
Stars
12
Forks
Watchers

A memory efficient DLRM training solution using ColossalAI

TensorNVMe

84
Stars
18
Forks
Watchers

A Python library transfers PyTorch tensors between CPU and NVMe

Classification

31
Stars
5
Forks
Watchers

视觉训练框架(简单 / 模块化 / 高扩展 / 分布式 / 自动剪枝)