colossal-ai

Here are 4 public repositories matching this topic...

A Python library transfers PyTorch tensors between CPU and NVMe

deep-learning pytorch nvme colossal-ai

A memory efficient DLRM training solution using ColossalAI

nlp deep-learning pytorch embeddings dlrm recommandation-system colossal-ai

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

pytorch classification face-recognition metric-learning mnn colossal-ai

Training LeNet-5 on MNIST using Colossal-AI

machine-learning mnist lenet-5 colossal-ai

Add a description, image, and links to the colossal-ai topic page so that developers can more easily learn about it.

To associate your repository with the colossal-ai topic, visit your repo's landing page and select "manage topics."