Path	Lines of Code
README.md	16
dags/examples/configs/__init__.py	1
dags/examples/configs/xpk_example_config.py	42
dags/examples/maxtext_aqtp_version_sweep_gke_example_dag.py	70
dags/examples/maxtext_sweep_gke_example_dag.py	32
dags/examples/xpk_example_dag.py	54
dags/legacy_test/templates/base.libsonnet	266
dags/legacy_test/templates/gpus.libsonnet	49
dags/legacy_test/templates/metrics.libsonnet	41
dags/legacy_test/templates/mixins.libsonnet	53
dags/legacy_test/templates/timeouts.libsonnet	22
dags/legacy_test/templates/tpus.libsonnet	76
dags/legacy_test/templates/utils.libsonnet	72
dags/legacy_test/templates/volumes.libsonnet	67
dags/legacy_test/tests/common.libsonnet	94
dags/legacy_test/tests/experimental.libsonnet	170
dags/legacy_test/tests/pytorch/common.libsonnet	97
dags/legacy_test/tests/pytorch/experimental.libsonnet	120
dags/legacy_test/tests/pytorch/nightly/accelerate-smoke.libsonnet	28
dags/legacy_test/tests/pytorch/nightly/ci.libsonnet	52
dags/legacy_test/tests/pytorch/nightly/common.libsonnet	207
dags/legacy_test/tests/pytorch/nightly/hf-bert.libsonnet	61
dags/legacy_test/tests/pytorch/nightly/llama2-model.libsonnet	226
dags/legacy_test/tests/pytorch/nightly/mnist.libsonnet	132
dags/legacy_test/tests/pytorch/nightly/resnet50-mp.libsonnet	232
dags/legacy_test/tests/pytorch/nightly/stable-diffusion-2.libsonnet	81
dags/legacy_test/tests/pytorch/r2.7/accelerate-smoke.libsonnet	28
dags/legacy_test/tests/pytorch/r2.7/ci.libsonnet	44
dags/legacy_test/tests/pytorch/r2.7/common.libsonnet	214
dags/legacy_test/tests/pytorch/r2.7/hf-bert.libsonnet	61
dags/legacy_test/tests/pytorch/r2.7/llama2-model.libsonnet	226
dags/legacy_test/tests/pytorch/r2.7/mnist.libsonnet	132
dags/legacy_test/tests/pytorch/r2.7/resnet50-mp.libsonnet	232
dags/legacy_test/tests/pytorch/r2.7/stable-diffusion-2.libsonnet	81
deployment/README.md	45
deployment/deployment_composer_env/README.md	21
deployment/schema/job_history.json	32
deployment/schema/metadata_history.json	20
deployment/schema/metric_history.json	20