Path	Lines of Code
README.MD	49
pyproject.toml	2
setup.py	16
torchrec/__init__.py	21
torchrec/datasets/__init__.py	4
torchrec/datasets/criteo.py	314
torchrec/datasets/experimental/__init__.py	1
torchrec/datasets/experimental/criteo_dataframes.py	72
torchrec/datasets/movielens.py	81
torchrec/datasets/random.py	127
torchrec/datasets/scripts/__init__.py	1
torchrec/datasets/scripts/preprocess_criteo.py	45
torchrec/datasets/utils.py	230
torchrec/distributed/__init__.py	20
torchrec/distributed/batched_embedding_kernel.py	613
torchrec/distributed/collective_utils.py	30
torchrec/distributed/comm.py	90
torchrec/distributed/comm_ops.py	616
torchrec/distributed/cw_sharding.py	139
torchrec/distributed/dist_data.py	415
torchrec/distributed/dp_sharding.py	192
torchrec/distributed/embedding.py	75
torchrec/distributed/embedding_kernel.py	275
torchrec/distributed/embedding_lookup.py	415
torchrec/distributed/embedding_sharding.py	376
torchrec/distributed/embedding_types.py	217
torchrec/distributed/embeddingbag.py	676
torchrec/distributed/grouped_position_weighted.py	84
torchrec/distributed/model_parallel.py	319
torchrec/distributed/planner/__init__.py	9
torchrec/distributed/planner/constants.py	39
torchrec/distributed/planner/enumerators.py	213
torchrec/distributed/planner/partitioners.py	265
torchrec/distributed/planner/perf_models.py	11
torchrec/distributed/planner/planners.py	260
torchrec/distributed/planner/proposers.py	111
torchrec/distributed/planner/shard_estimators.py	644
torchrec/distributed/planner/stats.py	209
torchrec/distributed/planner/storage_reservations.py	114
torchrec/distributed/planner/types.py	251
torchrec/distributed/planner/utils.py	11
torchrec/distributed/quant_embedding_kernel.py	159
torchrec/distributed/quant_embeddingbag.py	155
torchrec/distributed/rw_sharding.py	325
torchrec/distributed/train_pipeline.py	359
torchrec/distributed/tw_sharding.py	351
torchrec/distributed/twcw_sharding.py	21
torchrec/distributed/twrw_sharding.py	422
torchrec/distributed/types.py	265
torchrec/distributed/utils.py	77
torchrec/fx/__init__.py	1
torchrec/fx/tracer.py	24
torchrec/linter/module_linter.py	167
torchrec/models/__init__.py	2
torchrec/models/deepfm.py	121
torchrec/models/dlrm.py	134
torchrec/modules/__init__.py	9
torchrec/modules/activation.py	19
torchrec/modules/crossnet.py	222
torchrec/modules/deepfm.py	42
torchrec/modules/embedding_configs.py	56
torchrec/modules/embedding_modules.py	167
torchrec/modules/feature_processor.py	38
torchrec/modules/lazy_extension.py	112
torchrec/modules/mlp.py	79
torchrec/modules/score_learning.py	37
torchrec/modules/utils.py	77
torchrec/optim/__init__.py	13
torchrec/optim/clipping.py	28
torchrec/optim/fused.py	18
torchrec/optim/keyed.py	251
torchrec/optim/warmup.py	101
torchrec/quant/__init__.py	1
torchrec/quant/embedding_modules.py	197
torchrec/sparse/__init__.py	1
torchrec/sparse/jagged_tensor.py	819
torchrec/sparse/jagged_tensor_ops.cpp	314
torchrec/sparse/jagged_tensor_ops.h	62
torchrec/streamable.py	10