colossal-ai

There are 1 repository under colossal-ai topic.

hpcaitech / CachedEmbedding
A memory efficient DLRM training solution using ColossalAI
colossal-ai deep-learning dlrm embeddings nlp pytorch recommandation-system
Language:Python 90
hpcaitech / TensorNVMe
A Python library transfers PyTorch tensors between CPU and NVMe
colossal-ai deep-learning nvme pytorch
Language:C++ 81
bobo0810 / Classification
视觉训练框架(简单 / 模块化 / 高扩展 / 分布式 / 自动剪枝)
pytorch classification metric-learning colossal-ai face-recognition mnn
Language:Python 29
ctrlaltf2 / phonemes
An experiment in tokenization, grounded in linguistics
slm ai colossal-ai espeak-ng nlp tokenization
Language:C++ 0
manish-95 / LeNet-5-ColossalAI
Training LeNet-5 on MNIST using Colossal-AI
colossal-ai lenet-5 machine-learning mnist
Language:Jupyter Notebook 0

hpcaitech / CachedEmbedding