├── .gitignore ├── 00-dataset-management.ipynb ├── 01-dataset-preprocessing.ipynb ├── 02-model-training-hugectr.ipynb ├── 03-model-inference-hugectr.ipynb ├── 04-e2e-pipeline.ipynb ├── CONTRIBUTING ├── LICENSE ├── README.md ├── env-config └── Dockerfile ├── images ├── .gitkeep ├── bq_vertex_pipeline.png ├── dag_preprocessing.png ├── deepfm.png ├── dev-env.png ├── gcs_feat_vertex_pipeline.png ├── gcs_vertex_pipeline.png ├── gcsfuse.png ├── hugectrtrainer.png ├── merlin-vertex-e2e.png ├── merlin_components.png ├── merlin_vertexai.png ├── nvidia-merlin.png ├── nvt-pipeline.png ├── nvtabular_components.png ├── overview.png ├── preprocessing_pipeline_bq.png ├── preprocessing_pipeline_csv.png ├── triton-architecture.png ├── triton-in-vertex.png ├── triton-vertex.png └── vertexai_componentes.png └── src ├── Dockerfile.hugectr ├── Dockerfile.nvtabular ├── Dockerfile.triton ├── cloudbuild.yaml ├── pipelines ├── components.py ├── config.py ├── preprocessing_pipelines.py └── training_pipelines.py ├── preprocessing ├── __init__.py └── task.py ├── serving ├── __init__.py ├── entrypoint.sh ├── export.py ├── feature_utils.py └── inference.py └── training └── hugectr ├── __init__.py ├── model.py ├── task.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/.gitignore -------------------------------------------------------------------------------- /00-dataset-management.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/00-dataset-management.ipynb -------------------------------------------------------------------------------- /01-dataset-preprocessing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/01-dataset-preprocessing.ipynb -------------------------------------------------------------------------------- /02-model-training-hugectr.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/02-model-training-hugectr.ipynb -------------------------------------------------------------------------------- /03-model-inference-hugectr.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/03-model-inference-hugectr.ipynb -------------------------------------------------------------------------------- /04-e2e-pipeline.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/04-e2e-pipeline.ipynb -------------------------------------------------------------------------------- /CONTRIBUTING: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/CONTRIBUTING -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/README.md -------------------------------------------------------------------------------- /env-config/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/env-config/Dockerfile -------------------------------------------------------------------------------- /images/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /images/bq_vertex_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/bq_vertex_pipeline.png -------------------------------------------------------------------------------- /images/dag_preprocessing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/dag_preprocessing.png -------------------------------------------------------------------------------- /images/deepfm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/deepfm.png -------------------------------------------------------------------------------- /images/dev-env.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/dev-env.png -------------------------------------------------------------------------------- /images/gcs_feat_vertex_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/gcs_feat_vertex_pipeline.png -------------------------------------------------------------------------------- /images/gcs_vertex_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/gcs_vertex_pipeline.png -------------------------------------------------------------------------------- /images/gcsfuse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/gcsfuse.png -------------------------------------------------------------------------------- /images/hugectrtrainer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/hugectrtrainer.png -------------------------------------------------------------------------------- /images/merlin-vertex-e2e.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/merlin-vertex-e2e.png -------------------------------------------------------------------------------- /images/merlin_components.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/merlin_components.png -------------------------------------------------------------------------------- /images/merlin_vertexai.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/merlin_vertexai.png -------------------------------------------------------------------------------- /images/nvidia-merlin.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/nvidia-merlin.png -------------------------------------------------------------------------------- /images/nvt-pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/nvt-pipeline.png -------------------------------------------------------------------------------- /images/nvtabular_components.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/nvtabular_components.png -------------------------------------------------------------------------------- /images/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/overview.png -------------------------------------------------------------------------------- /images/preprocessing_pipeline_bq.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/preprocessing_pipeline_bq.png -------------------------------------------------------------------------------- /images/preprocessing_pipeline_csv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/preprocessing_pipeline_csv.png -------------------------------------------------------------------------------- /images/triton-architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/triton-architecture.png -------------------------------------------------------------------------------- /images/triton-in-vertex.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/triton-in-vertex.png -------------------------------------------------------------------------------- /images/triton-vertex.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/triton-vertex.png -------------------------------------------------------------------------------- /images/vertexai_componentes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/images/vertexai_componentes.png -------------------------------------------------------------------------------- /src/Dockerfile.hugectr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/Dockerfile.hugectr -------------------------------------------------------------------------------- /src/Dockerfile.nvtabular: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/Dockerfile.nvtabular -------------------------------------------------------------------------------- /src/Dockerfile.triton: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/Dockerfile.triton -------------------------------------------------------------------------------- /src/cloudbuild.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/cloudbuild.yaml -------------------------------------------------------------------------------- /src/pipelines/components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/pipelines/components.py -------------------------------------------------------------------------------- /src/pipelines/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/pipelines/config.py -------------------------------------------------------------------------------- /src/pipelines/preprocessing_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/pipelines/preprocessing_pipelines.py -------------------------------------------------------------------------------- /src/pipelines/training_pipelines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/pipelines/training_pipelines.py -------------------------------------------------------------------------------- /src/preprocessing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/preprocessing/__init__.py -------------------------------------------------------------------------------- /src/preprocessing/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/preprocessing/task.py -------------------------------------------------------------------------------- /src/serving/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/serving/entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/serving/entrypoint.sh -------------------------------------------------------------------------------- /src/serving/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/serving/export.py -------------------------------------------------------------------------------- /src/serving/feature_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/serving/feature_utils.py -------------------------------------------------------------------------------- /src/serving/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/serving/inference.py -------------------------------------------------------------------------------- /src/training/hugectr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/training/hugectr/__init__.py -------------------------------------------------------------------------------- /src/training/hugectr/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/training/hugectr/model.py -------------------------------------------------------------------------------- /src/training/hugectr/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/training/hugectr/task.py -------------------------------------------------------------------------------- /src/training/hugectr/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GoogleCloudPlatform/nvidia-merlin-on-vertex-ai/HEAD/src/training/hugectr/utils.py --------------------------------------------------------------------------------