Star AlbumentationsX on GitHub — it powers this leaderboard
hpcaitech/CachedEmbedding
A memory efficient DLRM training solution using ColossalAI