path	# lines of code
__init__.py	1
pyproject.toml	13
dags/inference/maxtext_model_config_generator.py	112
dags/inference/trtllm_bench_inference.py	28
dags/inference/configs/trtllm_bench_inference_config.py	99
dags/inference/configs/maxtext_inference_microbenchmark_gce_config.py	90
dags/inference/configs/trt_llm_mlperf_v40_config.py	127
dags/inference/configs/jetstream_benchmark_serving_gce_config.py	160
dags/inference/configs/trt_llm_mlperf_v41_config.py	190
dags/inference/configs/jetstream_pytorch_gce_config.py	96
dags/inference/configs/trt_llm_inference_config.py	139
dags/inference/utils/maxtext_gpu_microbenchmark_jsonl_converter.py	41
dags/inference/utils/trtllm_bench_jsonl_converter.py	65
dags/inference/trt_llm_mlperf_v41_inference.py	395
dags/inference/maxtext_inference_offline_benchmark.py	135
dags/inference/maxtext_inference_microbenchmark.py	271
dags/inference/trt_llm_mlperf_v40_inference.py	43
dags/inference/trt_llm_inference.py	27
dags/inference/maxtext_gpu_inference.py	28
dags/inference/jetstream_pytorch_inference.py	156
dags/inference/maxtext_inference.py	526
dags/inference/jetstream_inference_e2e.py	131
dags/pytorch_xla/pytorchxla2_torchbench.py	106
dags/pytorch_xla/configs/__init__.py	1
dags/pytorch_xla/configs/pytorchxla_torchbench_config.py	529
dags/pytorch_xla/nightly.py	231
dags/pytorch_xla/pytorchxla_torchbench.py	119
dags/pytorch_xla/r2_7.py	244
dags/pytorch_xla/pytorchxla-torchbench-release.py	112
dags/multipod/legacy_tests/gpt1-like.py	152
dags/multipod/maxtext_configs_aot.py	90
dags/multipod/maxtext_profiling_vertex_ai_tensorboard.py	53
dags/multipod/configs/jax_tests_gke_config.py	22
dags/multipod/configs/__init__.py	1
dags/multipod/configs/maxtext_sweep_gke_config.py	83
dags/multipod/configs/mxla_collective_config.py	62
dags/multipod/configs/gke_config.py	258
dags/multipod/configs/pytorch_config.py	25
dags/multipod/configs/common.py	24
dags/multipod/configs/maxtext_gce_config.py	114
dags/multipod/configs/jax_tests_gce_config.py	54
dags/multipod/maxtext_trillium_configs_perf.py	70
dags/multipod/maxtext_checkpointing.py	59
dags/multipod/mxla_collective_nightly.py	75
dags/multipod/maxtext_v5e_configs_perf.py	109
dags/multipod/maxtext_end_to_end.py	151
dags/multipod/maxtext_gpu_end_to_end.py	202
dags/multipod/maxtext_profiling.py	43
dags/multipod/maxtext_multi_tier_checkpointing.py	51
dags/multipod/legacy.py	130
dags/multipod/mxla_gpt3_6b_nightly_gke.py	97
dags/multipod/maxtext_sft_trainer.py	44
dags/multipod/maxtext_convergence.py	64
dags/multipod/maxtext_multi_tier_p2_checkpointing.py	80
dags/multipod/pytorch.py	48
dags/multipod/maxtext_configs_aot_hybridsim.py	92
dags/multipod/mxla_maxtext_nightly_gke.py	63
dags/common/vm_resource.py	285
dags/common/__init__.py	1
dags/common/model_configs.py	21
dags/__init__.py	1
dags/mantaray/run_mantaray_jobs.py	287
dags/mantaray/build_mantaray_docker.py	14
dags/composer_env.py	15
dags/map_reproducibility/utils/sample_workload_utils.py	245
dags/map_reproducibility/utils/file_comparison.py	88
dags/map_reproducibility/utils/common_utils.py	838
dags/map_reproducibility/utils/constants.py	37
dags/map_reproducibility/utils/internal_aotc_workload.py	178
dags/map_reproducibility/utils/benchmarkdb_utils.py	176
dags/map_reproducibility/internal_runs/cleanup_dags.py	29
dags/map_reproducibility/internal_runs/a3ultra_maxtext_benchmarking_dags.py	61
dags/map_reproducibility/internal_runs/dag_configs.py	197
dags/map_reproducibility/internal_runs/sample_a3ultra_maxtext_single_run.py	46
dags/map_reproducibility/internal_runs/sample_a4_maxtext_single_run.py	44
dags/map_reproducibility/internal_runs/a4_maxtext_benchmarking_dags.py	61
dags/map_reproducibility/internal_runs/backfill_dags.py	152
dags/map_reproducibility/internal_runs/sample_a3mega_maxtext_single_run.py	46
dags/map_reproducibility/internal_runs/a3mega_maxtext_benchmarking_dags.py	62
dags/map_reproducibility/a3ultra/nemo_two_node.py	40
dags/map_reproducibility/a3ultra/llama_3_1_70b_maxtext.py	63
dags/map_reproducibility/a3ultra/mixtral_8_7b_nemo.py	37
dags/map_reproducibility/a3ultra/llama_3_1_405b_nemo.py	44
dags/map_reproducibility/a3ultra/llama_3_1_405b_maxtext.py	46
dags/map_reproducibility/a3ultra/mixtral_8_7b_maxtext.py	50
dags/map_reproducibility/a3ultra/llama_3_1_70b_nemo.py	44
dags/map_reproducibility/a4/llama_3_1_405b/maxtext.py	46
dags/map_reproducibility/a4/llama_3_1_405b/nemo.py	37
dags/map_reproducibility/a4/two_node_nemo.py	43
dags/map_reproducibility/a4/mixtral_8_7b/nemo.py	37
dags/map_reproducibility/a4/llama3_1_70b/maxtext.py	44
dags/map_reproducibility/a4/llama3_1_70b/nemo.py	37
dags/map_reproducibility/a3mega/nemo_two_node.py	40
dags/map_reproducibility/a3mega/mixtral_8_7b_nemo.py	37
dags/map_reproducibility/a3mega/gpt3_175b_nemo.py	42
dags/map_reproducibility/a3mega/llama_3_70b_nemo.py	48
dags/map_reproducibility/a3mega/llama_3_1_70b_nemo.py	43
dags/framework3p/microbenchmarks_dag.py	83
dags/framework3p/configs/microbenchmarks_config.py	129
dags/solutions_team/solutionsteam_tf_nightly_supported.py	128
dags/solutions_team/solutionsteam_tf_release_se_supported.py	128
dags/solutions_team/configs/tensorflow/__init__.py	1
dags/solutions_team/configs/tensorflow/solutionsteam_tf_nightly_supported_config.py	491
dags/solutions_team/configs/tensorflow/solutionsteam_tf_release_supported_config.py	291
dags/solutions_team/configs/tensorflow/common.py	93
dags/solutions_team/configs/vllm/vllm_benchmark_config.py	265
dags/solutions_team/solutionsteam_tf_dlrm_benchmarks.py	108
dags/solutions_team/solutionsteam_tf_release_supported.py	95
dags/solutions_team/solutionsteam_vllm_benchmarks.py	107
dags/infra/clean_up.py	35
dags/gcs_bucket.py	7
dags/sparsity_diffusion_devx/maxtext_moe_gpu_e2e.py	61
dags/sparsity_diffusion_devx/project_bite_gpu_e2e.py	45
dags/sparsity_diffusion_devx/jax_stable_stack_tpu_e2e.py	114
dags/sparsity_diffusion_devx/configs/project_bite_config.py	114
dags/sparsity_diffusion_devx/configs/__init__.py	1
dags/sparsity_diffusion_devx/configs/gke_config.py	91
dags/sparsity_diffusion_devx/configs/common.py	16
dags/sparsity_diffusion_devx/maxtext_moe_tpu_e2e.py	135
dags/sparsity_diffusion_devx/jax_stable_stack_gpu_e2e.py	59
dags/sparsity_diffusion_devx/project_bite_tpu_e2e.py	76
dags/sparsity_diffusion_devx/maxdiffusion_e2e.py	68
dags/mlcompass/configs/simple_config.py	45
dags/mlcompass/simple_dag.py	19
dags/mlcompass/maxtext_gke.py	84
deployment/modules/composer_env/main.tf	47
deployment/modules/composer_env/variables.tf	10
deployment/provider.tf	10
deployment/bigquery.auto.tfvars	108
deployment/cloud_composer_template.tf	122
deployment/artifact_registry.tf	7
deployment/deployment_composer_env/provider.tf	4
deployment/deployment_composer_env/variables.tf	12
deployment/deployment_composer_env/composer_env.tf	6
deployment/clusters.tf	244
deployment/bigquery_template.tf	45
deployment/cloud_composer.auto.tfvars	15
xlml/utils/ssh.py	23
xlml/utils/mantaray.py	55
xlml/utils/bigquery.py	103
xlml/utils/metric.py	556
xlml/utils/xpk.py	231
xlml/utils/__init__.py	1
xlml/utils/gpu.py	381
xlml/utils/gke.py	146
xlml/utils/tpu.py	333
xlml/utils/composer.py	21
xlml/utils/name_format.py	25
xlml/utils/startup_script.py	6
xlml/apis/__init__.py	1
xlml/apis/metric_config.py	36
xlml/apis/xpk_cluster_config.py	9
xlml/apis/gcp_config.py	10
xlml/apis/task.py	541
xlml/__init__.py	1