Shortcuts

Python Module Index

t
 
t
torchrec
    torchrec.datasets
    torchrec.datasets.criteo
    torchrec.datasets.movielens
    torchrec.datasets.random
    torchrec.datasets.scripts
    torchrec.datasets.scripts.contiguous_preproc_criteo
    torchrec.datasets.scripts.npy_preproc_criteo
    torchrec.datasets.utils
    torchrec.distributed
    torchrec.distributed.collective_utils
    torchrec.distributed.comm
    torchrec.distributed.comm_ops
    torchrec.distributed.dist_data
    torchrec.distributed.embedding
    torchrec.distributed.embedding_lookup
    torchrec.distributed.embedding_sharding
    torchrec.distributed.embedding_types
    torchrec.distributed.embeddingbag
    torchrec.distributed.grouped_position_weighted
    torchrec.distributed.model_parallel
    torchrec.distributed.planner
    torchrec.distributed.planner.constants
    torchrec.distributed.planner.enumerators
    torchrec.distributed.planner.partitioners
    torchrec.distributed.planner.perf_models
    torchrec.distributed.planner.planners
    torchrec.distributed.planner.proposers
    torchrec.distributed.planner.shard_estimators
    torchrec.distributed.planner.stats
    torchrec.distributed.planner.storage_reservations
    torchrec.distributed.planner.types
    torchrec.distributed.planner.utils
    torchrec.distributed.quant_embeddingbag
    torchrec.distributed.sharding
    torchrec.distributed.sharding.cw_sharding
    torchrec.distributed.sharding.dp_sharding
    torchrec.distributed.sharding.rw_sharding
    torchrec.distributed.sharding.tw_sharding
    torchrec.distributed.sharding.twcw_sharding
    torchrec.distributed.sharding.twrw_sharding
    torchrec.distributed.train_pipeline
    torchrec.distributed.types
    torchrec.distributed.utils
    torchrec.fx
    torchrec.fx.tracer
    torchrec.models
    torchrec.models.deepfm
    torchrec.models.dlrm
    torchrec.modules
    torchrec.modules.activation
    torchrec.modules.crossnet
    torchrec.modules.deepfm
    torchrec.modules.embedding_configs
    torchrec.modules.embedding_modules
    torchrec.modules.feature_processor
    torchrec.modules.lazy_extension
    torchrec.modules.mlp
    torchrec.modules.utils
    torchrec.optim
    torchrec.optim.clipping
    torchrec.optim.fused
    torchrec.optim.keyed
    torchrec.optim.warmup
    torchrec.quant
    torchrec.quant.embedding_modules
    torchrec.sparse
    torchrec.sparse.jagged_tensor

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources