path	# lines of code
scripts/pyre_check.py	41
MANIFEST.in	1
setup.py	53
tzrec/metrics/recall_at_k.py	20
tzrec/metrics/__init__.py	1
tzrec/metrics/grouped_auc.py	105
tzrec/modules/sequence.py	359
tzrec/modules/interaction.py	54
tzrec/modules/__init__.py	1
tzrec/modules/extraction_net.py	102
tzrec/modules/masknet_module.py	92
tzrec/modules/dense_embedding_collection.py	195
tzrec/modules/embedding.py	1102
tzrec/modules/intervention.py	41
tzrec/modules/fm.py	11
tzrec/modules/activation.py	61
tzrec/modules/mlp.py	108
tzrec/modules/task_tower.py	24
tzrec/modules/variational_dropout.py	83
tzrec/modules/mmoe.py	49
tzrec/modules/utils.py	59
tzrec/modules/capsule.py	117
tzrec/modules/hstu.py	338
tzrec/predict.py	91
tzrec/constant.py	8
tzrec/optim/lr_scheduler.py	85
tzrec/optim/__init__.py	1
tzrec/optim/optimizer_builder.py	60
tzrec/utils/__init__.py	1
tzrec/utils/checkpoint_util.py	150
tzrec/utils/config_util.py	205
tzrec/utils/load_class.py	95
tzrec/utils/env_util.py	3
tzrec/utils/plan_util.py	196
tzrec/utils/logging_util.py	24
tzrec/utils/dist_util.py	216
tzrec/utils/faiss_util.py	86
tzrec/utils/misc_util.py	43
tzrec/utils/fx_util.py	108
tzrec/utils/state_dict_util.py	35
tzrec/models/multi_tower_din_trt.py	96
tzrec/models/dbmtl.py	120
tzrec/models/deepfm.py	63
tzrec/models/rank_model.py	311
tzrec/models/__init__.py	1
tzrec/models/dssm_v2.py	79
tzrec/models/mind.py	237
tzrec/models/ple.py	73
tzrec/models/dc2vr.py	121
tzrec/models/dat.py	174
tzrec/models/rocket_launching.py	224
tzrec/models/multi_tower_din.py	64
tzrec/models/tdm.py	90
tzrec/models/dssm.py	90
tzrec/models/mmoe.py	49
tzrec/models/multi_task_rank.py	136
tzrec/models/multi_tower.py	46
tzrec/models/masknet.py	32
tzrec/models/hstu.py	142
tzrec/models/dlrm.py	69
tzrec/models/match_model.py	270
tzrec/models/model.py	152
tzrec/tools/tdm/cluster_tree.py	98
tzrec/tools/tdm/retrieval.py	387
tzrec/tools/tdm/__init__.py	1
tzrec/tools/tdm/gen_tree/tree_generator.py	92
tzrec/tools/tdm/gen_tree/__init__.py	1
tzrec/tools/tdm/gen_tree/tree_cluster.py	230
tzrec/tools/tdm/gen_tree/tree_search_util.py	247
tzrec/tools/tdm/gen_tree/tree_builder.py	137
tzrec/tools/tdm/init_tree.py	91
tzrec/tools/create_faiss_index.py	70
tzrec/tools/__init__.py	1
tzrec/tools/create_fg_json.py	123
tzrec/tools/convert_easyrec_config_to_tzrec_config.py	689
tzrec/tools/feature_selection.py	235
tzrec/tools/hitrate.py	381
tzrec/tools/add_feature_info_to_config.py	214
tzrec/tools/create_online_infer_data.py	70
tzrec/tools/list_distcp_param.py	12
tzrec/__init__.py	22
tzrec/datasets/data_parser.py	609
tzrec/datasets/__init__.py	1
tzrec/datasets/sampler.py	705
tzrec/datasets/odps_dataset.py	504
tzrec/datasets/odps_dataset_v1.py	124
tzrec/datasets/parquet_dataset.py	197
tzrec/datasets/utils.py	291
tzrec/datasets/dataset.py	431
tzrec/datasets/csv_dataset.py	135
tzrec/export.py	36
tzrec/features/expr_feature.py	77
tzrec/features/__init__.py	1
tzrec/features/lookup_feature.py	210
tzrec/features/overlap_feature.py	85
tzrec/features/feature.py	659
tzrec/features/custom_feature.py	178
tzrec/features/raw_feature.py	97
tzrec/features/id_feature.py	146
tzrec/features/tokenize_feature.py	147
tzrec/features/match_feature.py	185
tzrec/features/combo_feature.py	96
tzrec/features/sequence_feature.py	529
tzrec/main.py	948
tzrec/protos/module.proto	67
tzrec/protos/eval.proto	8
tzrec/protos/data.proto	101
tzrec/protos/models/__init__.py	1
tzrec/protos/models/match_model.proto	69
tzrec/protos/models/general_rank_model.proto	12
tzrec/protos/models/rank_model.proto	32
tzrec/protos/models/multi_task_rank.proto	47
tzrec/protos/__init__.py	1
tzrec/protos/pipeline.proto	19
tzrec/protos/simi.proto	7
tzrec/protos/train.proto	28
tzrec/protos/seq_encoder.proto	78
tzrec/protos/tower.proto	141
tzrec/protos/model.proto	59
tzrec/protos/metric.proto	38
tzrec/protos/export.proto	4
tzrec/protos/optimizer.proto	136
tzrec/protos/feature.proto	711
tzrec/protos/loss.proto	20
tzrec/protos/sampler.proto	135
tzrec/train_eval.py	51
tzrec/acc/__init__.py	1
tzrec/acc/trt_utils.py	143
tzrec/acc/export_utils.py	182
tzrec/acc/aot_utils.py	9
tzrec/acc/utils.py	168
tzrec/benchmark/__init__.py	1
tzrec/benchmark/benchmark.py	261
tzrec/ops/hstu_compute.py	195
tzrec/ops/hstu_attention.py	141
tzrec/ops/benchmarks/hstu_attention_bench.py	277
tzrec/ops/__init__.py	6
tzrec/ops/pytorch/pt_hstu_linear.py	72
tzrec/ops/pytorch/pt_hstu_attention.py	200
tzrec/ops/pytorch/pt_layer_norm.py	38
tzrec/ops/pytorch/pt_jagged_tensors.py	214
tzrec/ops/pytorch/pt_position.py	117
tzrec/ops/position.py	112
tzrec/ops/layer_norm.py	59
tzrec/ops/utils.py	41
tzrec/ops/mm.py	13
tzrec/ops/jagged_tensors.py	187
tzrec/ops/triton/triton_hstu_linear.py	1186
tzrec/ops/triton/triton_jagged_tensors.py	631
tzrec/ops/triton/triton_position.py	586
tzrec/ops/triton/triton_hstu_attention.py	1640
tzrec/ops/triton/triton_hstu_preprocess_and_attention.py	307
tzrec/ops/triton/triton_addmm.py	281
tzrec/ops/triton/triton_layer_norm.py	733
tzrec/eval.py	36
tzrec/version.py	1
tzrec/loss/__init__.py	1
tzrec/loss/jrc_loss.py	64
setup.cfg	2