https://github.com/hpcaitech/CachedEmbedding

A memory efficient DLRM training solution using ColossalAI
https://github.com/hpcaitech/CachedEmbedding

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

Last synced: about 2 months ago
JSON representation

A memory efficient DLRM training solution using ColossalAI

• Host: GitHub
• URL: https://github.com/hpcaitech/CachedEmbedding
• Owner: hpcaitech
• License: apache-2.0
• Created: 2022-05-17T10:49:53.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-11-22T08:17:59.000Z (about 2 years ago)
• Last Synced: 2024-08-02T18:44:12.100Z (5 months ago)
• Topics: colossal-ai, deep-learning, dlrm, embeddings, nlp, pytorch, recommandation-system
• Language: Python
• Homepage:
• Size: 1.21 MB
• Stars: 92
• Watchers: 6
• Forks: 12
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: license

Awesome Lists containing this project