Add ZCH modules to TorchRec bento kernel

Summary: Can be used for faster prototyping in notebook

Reviewed By: spmex

Differential Revision: D79439896

Author: kausv

torchrec/__init__.py

@@ -21,6 +21,11 @@
     EmbeddingBagCollectionInterface,
     EmbeddingCollection,
 )  # noqa
+from torchrec.modules.hash_mc_modules import HashZchManagedCollisionModule  # noqa
+from torchrec.modules.mc_embedding_modules import (  # noqa
+    ManagedCollisionEmbeddingBagCollection,
+    ManagedCollisionEmbeddingCollection,
+)
 from torchrec.sparse.jagged_tensor import (  # noqa
     JaggedTensor,
     KeyedJaggedTensor,