.gitignore
.pre-commit-config.yaml
.pyre_configuration
.readthedocs.yaml
.ruff.toml
LICENSE
MANIFEST.in
README.md
pyproject.toml
requirements-cpu.txt
requirements-gpu.txt
requirements.txt
setup.cfg
setup.py
.github/workflows/benchmark.yml
.github/workflows/buildtest_ci.yml
.github/workflows/codestyle_ci.yml
.github/workflows/pytyping_ci.yml
.github/workflows/unittest_ci.yml
.github/workflows/unittest_cpu_ci.yml
.github/workflows/unittest_nightly.yml
.github/workflows/whl_and_doc_nightly.yml
data/.license_header.txt
data/test/id_vocab_dict_2
data/test/id_vocab_dict_3
data/test/id_vocab_list_0
data/test/id_vocab_list_1
data/test/spiece.model
data/test/tokenizer.json
docker/Dockerfile
docker/pip.conf
docs/Makefile
docs/make.bat
docs/images/intro.png
docs/images/models/dat.png
docs/images/models/dbmtl.png
docs/images/models/dbmtl_mmoe.png
docs/images/models/dcn_v1.jpg
docs/images/models/dcn_v2.jpg
docs/images/models/dcn_v2_cross.jpg
docs/images/models/deepfm.png
docs/images/models/din.png
docs/images/models/dlrm_hstu.png
docs/images/models/dssm_neg_sampler.png
docs/images/models/feature_groups_din.png
docs/images/models/masknet.jpg
docs/images/models/mind.png
docs/images/models/mmoe.png
docs/images/models/multi_tower.png
docs/images/models/ple.png
docs/images/models/rocket_launching.png
docs/images/models/tdm.png
docs/images/models/wide_and_deep.png
docs/images/qrcode/dinggroup1.png
docs/images/qrcode/dinggroup2.png
docs/source/conf.py
docs/source/develop.md
docs/source/faq.md
docs/source/index.rst
docs/source/quick_start.rst
docs/source/reference.md
docs/source/feature/autodis.md
docs/source/feature/data.md
docs/source/feature/dynamicemb.md
docs/source/feature/feature.md
docs/source/feature/zch.md
docs/source/models/dat.md
docs/source/models/dbmtl.md
docs/source/models/dcn.md
docs/source/models/dcn_v2.md
docs/source/models/deepfm.md
docs/source/models/din.md
docs/source/models/dlrm.md
docs/source/models/dlrm_hstu.md
docs/source/models/dssm.md
docs/source/models/feature_group.md
docs/source/models/generative.rst
docs/source/models/loss.md
docs/source/models/masknet.md
docs/source/models/mind.md
docs/source/models/mmoe.md
docs/source/models/multi_target.rst
docs/source/models/multi_tower.md
docs/source/models/optimizer.md
docs/source/models/ple.md
docs/source/models/rank.rst
docs/source/models/recall.rst
docs/source/models/rocket_launching.md
docs/source/models/tdm.md
docs/source/models/user_define.md
docs/source/models/wide_and_deep.md
docs/source/models/xdeepfm.md
docs/source/quick_start/dlc_odps_dataset_tutorial.md
docs/source/quick_start/dlc_tutorial.md
docs/source/quick_start/local_tutorial.md
docs/source/quick_start/local_tutorial_tdm.md
docs/source/quick_start/local_tutorial_u2i_vec.md
docs/source/usage/convert_easyrec_config_to_tzrec_config.md
docs/source/usage/eval.md
docs/source/usage/export.md
docs/source/usage/feature_selection.md
docs/source/usage/predict.md
docs/source/usage/serving.md
docs/source/usage/train.md
examples/dbmtl_taobao.config
examples/dbmtl_taobao_jrc.config
examples/dbmtl_taobao_seq.config
examples/deepfm_criteo.config
examples/dlrm_criteo.config
examples/dssm_taobao.config
examples/dssm_v2_taobao.config
examples/masknet_criteo.config
examples/mind_taobao.config
examples/mmoe_taobao.config
examples/multi_tower_din_taobao.config
examples/multi_tower_taobao.config
examples/ple_taobao.config
examples/rocket_launching_criteo.config
examples/tdm_taobao.config
requirements/cpu.txt
requirements/docs.txt
requirements/gpu.txt
requirements/runtime.txt
requirements/test.txt
scripts/build_docker.sh
scripts/build_wheel.sh
scripts/gen_proto.sh
scripts/pyre_check.py
scripts/ci/ci_data.sh
scripts/ci/ci_test.sh
scripts/ci/ci_test_cpu.sh
scripts/doc/build_doc_pre_work.sh
scripts/doc/build_docs.sh
tzrec/__init__.py
tzrec/constant.py
tzrec/eval.py
tzrec/export.py
tzrec/main.py
tzrec/predict.py
tzrec/train_eval.py
tzrec/version.py
tzrec/acc/__init__.py
tzrec/acc/aot_utils.py
tzrec/acc/pt2_utils.py
tzrec/acc/trt_utils.py
tzrec/acc/utils.py
tzrec/benchmark/__init__.py
tzrec/benchmark/benchmark.py
tzrec/benchmark/configs/base_eval_metric.json
tzrec/benchmark/configs/criteo/deepfm.config
tzrec/benchmark/configs/criteo/dlrm.config
tzrec/benchmark/configs/criteo/rocket_launching.config
tzrec/benchmark/configs/kuairand/dlrm_hstu.config
tzrec/benchmark/configs/taobao/dbmtl.config
tzrec/benchmark/configs/taobao/dbmtl_has_sequence.config
tzrec/benchmark/configs/taobao/dbmtl_jrc.config
tzrec/benchmark/configs/taobao/dssm_taobao.config
tzrec/benchmark/configs/taobao/mmoe.config
tzrec/benchmark/configs/taobao/mmoe_has_sequence.config
tzrec/benchmark/configs/taobao/ple.config
tzrec/benchmark/configs/taobao/ple_has_sequence.config
tzrec/benchmark/configs/taobao_ccp/dbmtl.config
tzrec/benchmark/configs/taobao_ccp/mmoe.config
tzrec/benchmark/configs/taobao_ccp/ple.config
tzrec/datasets/__init__.py
tzrec/datasets/csv_dataset.py
tzrec/datasets/csv_dataset_test.py
tzrec/datasets/data_parser.py
tzrec/datasets/data_parser_test.py
tzrec/datasets/dataset.py
tzrec/datasets/dataset_test.py
tzrec/datasets/odps_dataset.py
tzrec/datasets/odps_dataset_test.py
tzrec/datasets/odps_dataset_v1.py
tzrec/datasets/odps_dataset_v1_test.py
tzrec/datasets/parquet_dataset.py
tzrec/datasets/parquet_dataset_test.py
tzrec/datasets/sampler.py
tzrec/datasets/sampler_test.py
tzrec/datasets/utils.py
tzrec/datasets/utils_test.py
tzrec/features/__init__.py
tzrec/features/bool_mask_feature.py
tzrec/features/bool_mask_feature_test.py
tzrec/features/combo_feature.py
tzrec/features/combo_feature_test.py
tzrec/features/custom_feature.py
tzrec/features/custom_feature_test.py
tzrec/features/expr_feature.py
tzrec/features/expr_feature_test.py
tzrec/features/feature.py
tzrec/features/feature_test.py
tzrec/features/id_feature.py
tzrec/features/id_feature_test.py
tzrec/features/kv_dot_product.py
tzrec/features/kv_dot_product_test.py
tzrec/features/lookup_feature.py
tzrec/features/lookup_feature_test.py
tzrec/features/match_feature.py
tzrec/features/match_feature_test.py
tzrec/features/overlap_feature.py
tzrec/features/overlap_feature_test.py
tzrec/features/raw_feature.py
tzrec/features/raw_feature_test.py
tzrec/features/sequence_feature.py
tzrec/features/sequence_feature_test.py
tzrec/features/tokenize_feature.py
tzrec/features/tokenize_feature_test.py
tzrec/inference/__init__.py
tzrec/inference/api.py
tzrec/loss/__init__.py
tzrec/loss/focal_loss.py
tzrec/loss/focal_loss_test.py
tzrec/loss/jrc_loss.py
tzrec/loss/jrc_loss_test.py
tzrec/metrics/__init__.py
tzrec/metrics/grouped_auc.py
tzrec/metrics/grouped_auc_test.py
tzrec/metrics/grouped_xauc.py
tzrec/metrics/grouped_xauc_test.py
tzrec/metrics/recall_at_k.py
tzrec/metrics/recall_at_k_test.py
tzrec/metrics/xauc.py
tzrec/metrics/xauc_test.py
tzrec/models/__init__.py
tzrec/models/dat.py
tzrec/models/dat_test.py
tzrec/models/dbmtl.py
tzrec/models/dbmtl_test.py
tzrec/models/dc2vr.py
tzrec/models/dc2vr_test.py
tzrec/models/dcn.py
tzrec/models/dcn_test.py
tzrec/models/dcn_v2.py
tzrec/models/dcn_v2_test.py
tzrec/models/deepfm.py
tzrec/models/deepfm_test.py
tzrec/models/dlrm.py
tzrec/models/dlrm_hstu.py
tzrec/models/dlrm_hstu_test.py
tzrec/models/dlrm_test.py
tzrec/models/dssm.py
tzrec/models/dssm_test.py
tzrec/models/dssm_v2.py
tzrec/models/dssm_v2_test.py
tzrec/models/hstu.py
tzrec/models/hstu_test.py
tzrec/models/masknet.py
tzrec/models/masknet_test.py
tzrec/models/match_model.py
tzrec/models/match_model_test.py
tzrec/models/mind.py
tzrec/models/mind_test.py
tzrec/models/mmoe.py
tzrec/models/mmoe_test.py
tzrec/models/model.py
tzrec/models/multi_task_rank.py
tzrec/models/multi_task_rank_test.py
tzrec/models/multi_tower.py
tzrec/models/multi_tower_din.py
tzrec/models/multi_tower_din_test.py
tzrec/models/multi_tower_din_trt.py
tzrec/models/multi_tower_test.py
tzrec/models/ple.py
tzrec/models/ple_test.py
tzrec/models/rank_model.py
tzrec/models/rank_model_test.py
tzrec/models/rocket_launching.py
tzrec/models/rocket_launching_test.py
tzrec/models/tdm.py
tzrec/models/tdm_test.py
tzrec/models/wide_and_deep.py
tzrec/models/wide_and_deep_test.py
tzrec/models/xdeepfm.py
tzrec/models/xdeepfm_test.py
tzrec/modules/__init__.py
tzrec/modules/activation.py
tzrec/modules/activation_test.py
tzrec/modules/capsule.py
tzrec/modules/capsule_test.py
tzrec/modules/dense_embedding_collection.py
tzrec/modules/dense_embedding_collection_test.py
tzrec/modules/embedding.py
tzrec/modules/embedding_test.py
tzrec/modules/extraction_net.py
tzrec/modules/extraction_net_test.py
tzrec/modules/fm.py
tzrec/modules/fm_test.py
tzrec/modules/hstu.py
tzrec/modules/interaction.py
tzrec/modules/interaction_test.py
tzrec/modules/intervention.py
tzrec/modules/intervention_test.py
tzrec/modules/masknet.py
tzrec/modules/masknet_test.py
tzrec/modules/mlp.py
tzrec/modules/mlp_test.py
tzrec/modules/mmoe.py
tzrec/modules/mmoe_test.py
tzrec/modules/norm.py
tzrec/modules/norm_test.py
tzrec/modules/sequence.py
tzrec/modules/sequence_test.py
tzrec/modules/task_tower.py
tzrec/modules/task_tower_test.py
tzrec/modules/utils.py
tzrec/modules/utils_test.py
tzrec/modules/variational_dropout.py
tzrec/modules/variational_dropout_test.py
tzrec/modules/gr/__init__.py
tzrec/modules/gr/action_encoder.py
tzrec/modules/gr/action_encoder_test.py
tzrec/modules/gr/content_encoder.py
tzrec/modules/gr/content_encoder_test.py
tzrec/modules/gr/contextualize_mlps.py
tzrec/modules/gr/hstu_transducer.py
tzrec/modules/gr/positional_encoder.py
tzrec/modules/gr/postprocessors.py
tzrec/modules/gr/postprocessors_test.py
tzrec/modules/gr/preprocessors.py
tzrec/modules/gr/preprocessors_test.py
tzrec/modules/gr/stu.py
tzrec/modules/gr/stu_test.py
tzrec/ops/__init__.py
tzrec/ops/hstu_attention.py
tzrec/ops/hstu_attention_test.py
tzrec/ops/hstu_compute.py
tzrec/ops/hstu_compute_test.py
tzrec/ops/jagged_tensors.py
tzrec/ops/jagged_tensors_test.py
tzrec/ops/layer_norm.py
tzrec/ops/layer_norm_test.py
tzrec/ops/mm.py
tzrec/ops/mm_test.py
tzrec/ops/position.py
tzrec/ops/position_test.py
tzrec/ops/utils.py
tzrec/ops/utils_test.py
tzrec/ops/benchmarks/hstu_attention_bench.py
tzrec/ops/pytorch/__init__.py
tzrec/ops/pytorch/pt_hstu_attention.py
tzrec/ops/pytorch/pt_hstu_linear.py
tzrec/ops/pytorch/pt_jagged_tensors.py
tzrec/ops/pytorch/pt_layer_norm.py
tzrec/ops/pytorch/pt_position.py
tzrec/ops/triton/__init__.py
tzrec/ops/triton/triton_addmm.py
tzrec/ops/triton/triton_hstu_attention.py
tzrec/ops/triton/triton_hstu_linear.py
tzrec/ops/triton/triton_hstu_preprocess_and_attention.py
tzrec/ops/triton/triton_jagged_tensors.py
tzrec/ops/triton/triton_layer_norm.py
tzrec/ops/triton/triton_position.py
tzrec/optim/__init__.py
tzrec/optim/lr_scheduler.py
tzrec/optim/lr_scheduler_test.py
tzrec/optim/optimizer.py
tzrec/optim/optimizer_builder.py
tzrec/optim/optimizer_builder_test.py
tzrec/optim/optimizer_test.py
tzrec/protos/__init__.py
tzrec/protos/data.proto
tzrec/protos/data_pb2.py
tzrec/protos/data_pb2.pyi
tzrec/protos/eval.proto
tzrec/protos/eval_pb2.py
tzrec/protos/eval_pb2.pyi
tzrec/protos/export.proto
tzrec/protos/export_pb2.py
tzrec/protos/export_pb2.pyi
tzrec/protos/feature.proto
tzrec/protos/feature_pb2.py
tzrec/protos/feature_pb2.pyi
tzrec/protos/loss.proto
tzrec/protos/loss_pb2.py
tzrec/protos/loss_pb2.pyi
tzrec/protos/metric.proto
tzrec/protos/metric_pb2.py
tzrec/protos/metric_pb2.pyi
tzrec/protos/model.proto
tzrec/protos/model_pb2.py
tzrec/protos/model_pb2.pyi
tzrec/protos/module.proto
tzrec/protos/module_pb2.py
tzrec/protos/module_pb2.pyi
tzrec/protos/optimizer.proto
tzrec/protos/optimizer_pb2.py
tzrec/protos/optimizer_pb2.pyi
tzrec/protos/pipeline.proto
tzrec/protos/pipeline_pb2.py
tzrec/protos/pipeline_pb2.pyi
tzrec/protos/sampler.proto
tzrec/protos/sampler_pb2.py
tzrec/protos/sampler_pb2.pyi
tzrec/protos/seq_encoder.proto
tzrec/protos/seq_encoder_pb2.py
tzrec/protos/seq_encoder_pb2.pyi
tzrec/protos/simi.proto
tzrec/protos/simi_pb2.py
tzrec/protos/simi_pb2.pyi
tzrec/protos/tower.proto
tzrec/protos/tower_pb2.py
tzrec/protos/tower_pb2.pyi
tzrec/protos/train.proto
tzrec/protos/train_pb2.py
tzrec/protos/train_pb2.pyi
tzrec/protos/models/__init__.py
tzrec/protos/models/general_rank_model.proto
tzrec/protos/models/general_rank_model_pb2.py
tzrec/protos/models/general_rank_model_pb2.pyi
tzrec/protos/models/match_model.proto
tzrec/protos/models/match_model_pb2.py
tzrec/protos/models/match_model_pb2.pyi
tzrec/protos/models/multi_task_rank.proto
tzrec/protos/models/multi_task_rank_pb2.py
tzrec/protos/models/multi_task_rank_pb2.pyi
tzrec/protos/models/rank_model.proto
tzrec/protos/models/rank_model_pb2.py
tzrec/protos/models/rank_model_pb2.pyi
tzrec/tests/__init__.py
tzrec/tests/match_integration_test.py
tzrec/tests/rank_integration_test.py
tzrec/tests/run.py
tzrec/tests/utils.py
tzrec/tests/configs/dat_mock.config
tzrec/tests/configs/dbmtl_has_sequence_mock.config
tzrec/tests/configs/dbmtl_has_sequence_mock_focal_loss.config
tzrec/tests/configs/dbmtl_has_sequence_variational_dropout_mock.config
tzrec/tests/configs/dlrm_hstu_kuairand_1k.config
tzrec/tests/configs/dssm_fg_hard_negative_mock.config
tzrec/tests/configs/dssm_fg_mock.config
tzrec/tests/configs/dssm_mock.config
tzrec/tests/configs/dssm_v2_fg_mock.config
tzrec/tests/configs/dssm_v2_mlpemb_fg_mock.config
tzrec/tests/configs/dssm_variational_dropout_mock.config
tzrec/tests/configs/hstu_fg_mock.config
tzrec/tests/configs/mind_mock.config
tzrec/tests/configs/multi_tower_din_fg_dynamicemb_mock.config
tzrec/tests/configs/multi_tower_din_fg_mock.config
tzrec/tests/configs/multi_tower_din_fg_mock_bf16_emb_fp16.config
tzrec/tests/configs/multi_tower_din_fg_mock_fp16_ga.config
tzrec/tests/configs/multi_tower_din_mock.config
tzrec/tests/configs/multi_tower_din_mock_frozen.config
tzrec/tests/configs/multi_tower_din_trt_fg_mock.config
tzrec/tests/configs/multi_tower_din_zch_fg_mock.config
tzrec/tests/configs/multi_tower_din_zch_trt_fg_mock.config
tzrec/tests/configs/tdm_fg_mock.config
tzrec/tools/__init__.py
tzrec/tools/add_feature_info_to_config.py
tzrec/tools/add_feature_info_to_config_test.py
tzrec/tools/convert_easyrec_config_to_tzrec_config.py
tzrec/tools/convert_easyrec_config_to_tzrec_config_test.py
tzrec/tools/create_faiss_index.py
tzrec/tools/create_fg_json.py
tzrec/tools/create_online_infer_data.py
tzrec/tools/feature_selection.py
tzrec/tools/hitrate.py
tzrec/tools/list_distcp_param.py
tzrec/tools/dynamicemb/__init__.py
tzrec/tools/dynamicemb/create_dynamicemb_init_ckpt.py
tzrec/tools/dynamicemb/create_dynamicemb_init_ckpt_test.py
tzrec/tools/tdm/__init__.py
tzrec/tools/tdm/cluster_tree.py
tzrec/tools/tdm/init_tree.py
tzrec/tools/tdm/retrieval.py
tzrec/tools/tdm/gen_tree/__init__.py
tzrec/tools/tdm/gen_tree/tree_builder.py
tzrec/tools/tdm/gen_tree/tree_builder_test.py
tzrec/tools/tdm/gen_tree/tree_cluster.py
tzrec/tools/tdm/gen_tree/tree_cluster_test.py
tzrec/tools/tdm/gen_tree/tree_generator.py
tzrec/tools/tdm/gen_tree/tree_generator_test.py
tzrec/tools/tdm/gen_tree/tree_search_util.py
tzrec/tools/tdm/gen_tree/tree_search_util_test.py
tzrec/utils/__init__.py
tzrec/utils/checkpoint_util.py
tzrec/utils/checkpoint_util_test.py
tzrec/utils/config_util.py
tzrec/utils/config_util_test.py
tzrec/utils/dist_util.py
tzrec/utils/dynamicemb_util.py
tzrec/utils/env_util.py
tzrec/utils/export_util.py
tzrec/utils/faiss_util.py
tzrec/utils/faiss_util_test.py
tzrec/utils/fx_util.py
tzrec/utils/load_class.py
tzrec/utils/load_class_test.py
tzrec/utils/logging_util.py
tzrec/utils/misc_util.py
tzrec/utils/plan_util.py
tzrec/utils/plan_util_test.py
tzrec/utils/state_dict_util.py
tzrec/utils/test_util.py
tzrec_sekai.egg-info/PKG-INFO
tzrec_sekai.egg-info/SOURCES.txt
tzrec_sekai.egg-info/dependency_links.txt
tzrec_sekai.egg-info/top_level.txt