/CachedEmbedding

A memory efficient DLRM training solution using ColossalAI

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers