Skip to content

Actions: NVIDIA/JAX-Toolbox

Actions

nsys-jax pure-Python CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
nsys-jax.yaml
638 workflow runs
638 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Update the dockerfile base image to cuda-dl-base nsys-jax pure-Python CI #577: Pull request #1248 synchronize by Steboss
January 15, 2025 09:36 1m 40s sbosisio/cuda-dl-base
sbosisio/cuda-dl-base
January 15, 2025 09:36 1m 40s
Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs (#1249) nsys-jax pure-Python CI #576: Commit 2e25d13 pushed by olupton
January 15, 2025 08:41 1m 34s main
main
January 15, 2025 08:41 1m 34s
Delete .git for internal repos nsys-jax pure-Python CI #575: Pull request #1252 opened by DwarKapex
January 15, 2025 04:13 1m 40s 25.01-devel-clean-internal-git
25.01-devel-clean-internal-git
January 15, 2025 04:13 1m 40s
Update ml_dtypes to the latest version after pip-finalize. nsys-jax pure-Python CI #574: Pull request #1250 opened by DwarKapex
January 15, 2025 00:27 1m 39s 25.01-devel-ml-dtypes-050
25.01-devel-ml-dtypes-050
January 15, 2025 00:27 1m 39s
Remove V100 from test environment (#1238) nsys-jax pure-Python CI #573: Commit 505b741 pushed by yhtang
January 14, 2025 18:14 1m 39s main
main
January 14, 2025 18:14 1m 39s
Update the dockerfile base image to cuda-dl-base nsys-jax pure-Python CI #572: Pull request #1248 synchronize by Steboss
January 14, 2025 15:33 2m 18s sbosisio/cuda-dl-base
sbosisio/cuda-dl-base
January 14, 2025 15:33 2m 18s
Only set CUDA_DEVICE_MAX_CONNECTIONS=1 for Hopper/cc9.0 runs nsys-jax pure-Python CI #571: Pull request #1249 opened by olupton
January 14, 2025 12:00 1m 40s olupton/cuda-device-max-connections
olupton/cuda-device-max-connections
January 14, 2025 12:00 1m 40s
Update the dockerfile base image to cuda-dl-base nsys-jax pure-Python CI #570: Pull request #1248 opened by Steboss
January 14, 2025 11:42 1m 38s sbosisio/cuda-dl-base
sbosisio/cuda-dl-base
January 14, 2025 11:42 1m 38s
Remove V100 from test environment nsys-jax pure-Python CI #569: Pull request #1238 synchronize by DwarKapex
January 13, 2025 19:45 1m 33s vkozlov/remove-v100
vkozlov/remove-v100
January 13, 2025 19:45 1m 33s
[nsys-jax] Add ratio of hidden communication time to total communication time nsys-jax pure-Python CI #568: Pull request #1241 synchronize by sfvaroglu
January 13, 2025 18:47 1m 36s sevin/comm_time
sevin/comm_time
January 13, 2025 18:47 1m 36s
[nsys-jax] Add ratio of hidden communication time to total communication time nsys-jax pure-Python CI #567: Pull request #1241 synchronize by sfvaroglu
January 13, 2025 18:44 1m 38s sevin/comm_time
sevin/comm_time
January 13, 2025 18:44 1m 38s
[nsys-jax] Add ratio of hidden communication time to total communication time nsys-jax pure-Python CI #566: Pull request #1241 synchronize by sfvaroglu
January 13, 2025 18:30 1m 34s sevin/comm_time
sevin/comm_time
January 13, 2025 18:30 1m 34s
[nsys-jax] Add ratio of hidden communication time to total communication time nsys-jax pure-Python CI #565: Pull request #1241 synchronize by sfvaroglu
January 13, 2025 18:15 4m 3s sevin/comm_time
sevin/comm_time
January 13, 2025 18:15 4m 3s
CI: run MaxText tests on AWS with NGC release candidate images nsys-jax pure-Python CI #564: Pull request #1237 synchronize by olupton
January 13, 2025 17:35 1m 36s olupton/eks-maxtext-25.01
olupton/eks-maxtext-25.01
January 13, 2025 17:35 1m 36s
Add MACE training example nsys-jax pure-Python CI #563: Pull request #1192 synchronize by mariogeiger
January 12, 2025 21:26 1m 33s mariogeiger:mace
mariogeiger:mace
January 12, 2025 21:26 1m 33s
Replace deprecated flag xla_gpu_graph_level. nsys-jax pure-Python CI #562: Pull request #1244 opened by sergachev
January 11, 2025 21:56 1m 37s cuda_graph_flag
cuda_graph_flag
January 11, 2025 21:56 1m 37s
Remove V100 from test environment nsys-jax pure-Python CI #561: Pull request #1238 synchronize by DwarKapex
January 10, 2025 22:30 1m 35s vkozlov/remove-v100
vkozlov/remove-v100
January 10, 2025 22:30 1m 35s
Remove V100 from test environment nsys-jax pure-Python CI #560: Pull request #1238 synchronize by DwarKapex
January 10, 2025 21:57 1m 35s vkozlov/remove-v100
vkozlov/remove-v100
January 10, 2025 21:57 1m 35s
Remove V100 from test environment nsys-jax pure-Python CI #559: Pull request #1238 synchronize by DwarKapex
January 10, 2025 21:57 18s vkozlov/remove-v100
vkozlov/remove-v100
January 10, 2025 21:57 18s
Remove V100 from test environment nsys-jax pure-Python CI #558: Pull request #1238 synchronize by DwarKapex
January 10, 2025 21:57 40s vkozlov/remove-v100
vkozlov/remove-v100
January 10, 2025 21:57 40s
[nsys-jax] Add ratio of hidden communication time to total communication time nsys-jax pure-Python CI #557: Pull request #1241 opened by sfvaroglu
January 10, 2025 18:07 1m 33s sevin/comm_time
sevin/comm_time
January 10, 2025 18:07 1m 33s
fix t5x tests (#1208) nsys-jax pure-Python CI #556: Commit 34066f5 pushed by ashors1
January 10, 2025 17:59 1m 31s main
main
January 10, 2025 17:59 1m 31s
Add JetStream to MaxText container nsys-jax pure-Python CI #555: Pull request #1058 synchronize by DwarKapex
January 10, 2025 16:58 6m 6s vkozlov/jetstream-4-maxtext
vkozlov/jetstream-4-maxtext
January 10, 2025 16:58 6m 6s
No need to build tensorflow-text from source: py3.12 x86/arm64 wheels… nsys-jax pure-Python CI #554: Commit 26451c0 pushed by DwarKapex
January 10, 2025 16:41 1m 39s main
main
January 10, 2025 16:41 1m 39s
Remove deprecated flag xla_gpu_enable_triton_softmax_fusion. nsys-jax pure-Python CI #553: Pull request #1240 opened by sergachev
January 10, 2025 16:23 1m 35s triton_softmax_fusion_flag
triton_softmax_fusion_flag
January 10, 2025 16:23 1m 35s