torchrec/distributed/comm_ops.py (2 lines):
	- line 34: # TODO: T96382816, NE Parity Backward compatibility
	- line 412: # TODO: improve performance of _recat_pooled_embedding_grad_out and


torchrec/datasets/utils.py (2 lines):
	- line 234: TODO:
	- line 252: # TODO: enforce typing for each instance based on mode, otherwise


torchrec/distributed/dist_data.py (1 line):
	- line 505: # TODO: potentially refactor to take KT instead of torch.Tensor: D29174501


torchrec/datasets/experimental/criteo_dataframes.py (1 line):
	- line 59: # TODO: Fix safe_cast type annotation


torchrec/datasets/random.py (1 line):
	- line 132: (RecSys). Currently produces unweighted sparse features only. TODO: Add


torchrec/distributed/embeddingbag.py (1 line):
	- line 632: # TODO: enable it with correct semantics, see T104397332


torchrec/distributed/planner/shard_estimators.py (1 line):
	- line 371: # TODO: this is allreduce perf, better separated out as backward perf