We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d1a4303 commit dbeadccCopy full SHA for dbeadcc
conf/common/test/dse_jaxtoolbox_grok.toml
@@ -19,7 +19,7 @@ description = "DSE JaxToolbox Grok"
19
test_template_name = "JaxToolboxGrok"
20
21
[cmd_args]
22
-docker_image_url = "https://docker/url"
+docker_image_url = "ghcr.io/nvidia/jax:pax-2025-01-09"
23
[cmd_args.fdl]
24
num_gpus = [1, 8, 16]
25
checkpoint_policy = ["save_iteration_input", "save_none"]
0 commit comments