Is TritonBatchedFusedEmbeddingBag supported in open-source TorchRec?

[Pluto1314]

Hi,

I noticed that TritonBatchedFusedEmbeddingBag (a new Triton-based batched fused embedding bag implementation) was introduced in the v1.6.0 release as an alternative to FBGEMM TBE.

I found an internal import path that seems to use this implementation:

from deeplearning.fbgemm.fbgemm_gpu.fb.triton.triton_table_batched_embeddings import (
    TritonTableBatchedEmbeddingBags,
)

Is the module at deeplearning.fbgemm.fbgemm_gpu.fb.triton.triton_table_batched_embeddings an open-source repository, or is it Meta-internal only? I noticed the fb directory in the path and couldn't find it in the public FBGEMM or TorchRec repos.