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
250 workflow runs
250 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Adds patch to rosetta-pax for PP fix nsys-jax pure-Python CI #155: Pull request #981 synchronize by terrykong
August 9, 2024 17:18 1m 27s add-kaixi-pp-patch
add-kaixi-pp-patch
August 9, 2024 17:18 1m 27s
nsys-jax: support fp8 and new XLA proto formats nsys-jax pure-Python CI #154: Pull request #990 opened by olupton
August 9, 2024 14:54 1m 24s olupton/xla-collective-annotation-changes
olupton/xla-collective-annotation-changes
August 9, 2024 14:54 1m 24s
fix remat policy and fused attn variable in test-maxtext.sh (#982) nsys-jax pure-Python CI #153: Commit 1fd097d pushed by olupton
August 9, 2024 12:51 1m 47s main
main
August 9, 2024 12:51 1m 47s
nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of sin… nsys-jax pure-Python CI #152: Commit 15e65e9 pushed by olupton
August 9, 2024 12:31 1m 27s main
main
August 9, 2024 12:31 1m 27s
nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better nsys-jax pure-Python CI #151: Pull request #988 synchronize by olupton
August 9, 2024 10:31 1m 32s olupton/nsys-jax-single-gpu
olupton/nsys-jax-single-gpu
August 9, 2024 10:31 1m 32s
nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better nsys-jax pure-Python CI #150: Pull request #988 synchronize by olupton
August 9, 2024 10:30 1m 27s olupton/nsys-jax-single-gpu
olupton/nsys-jax-single-gpu
August 9, 2024 10:30 1m 27s
nsys-jax: roll back to 2024.4.1 and handle single-gpu profiles of single module executions better nsys-jax pure-Python CI #149: Pull request #988 opened by olupton
August 9, 2024 10:25 1m 13s olupton/nsys-jax-single-gpu
olupton/nsys-jax-single-gpu
August 9, 2024 10:25 1m 13s
Fix for jax build including cuda plugin (#987) nsys-jax pure-Python CI #148: Commit 955177d pushed by olupton
August 9, 2024 09:05 1m 25s main
main
August 9, 2024 09:05 1m 25s
Fix for jax build including cuda plugin nsys-jax pure-Python CI #147: Pull request #987 synchronize by DwarKapex
August 9, 2024 06:53 1m 31s jax-add-cuda-plugin
jax-add-cuda-plugin
August 9, 2024 06:53 1m 31s
Fix for jax build including cuda plugin nsys-jax pure-Python CI #146: Pull request #987 synchronize by DwarKapex
August 9, 2024 05:28 1m 27s jax-add-cuda-plugin
jax-add-cuda-plugin
August 9, 2024 05:28 1m 27s
Fix for jax build including cuda plugin nsys-jax pure-Python CI #145: Pull request #987 synchronize by DwarKapex
August 9, 2024 04:41 1m 28s jax-add-cuda-plugin
jax-add-cuda-plugin
August 9, 2024 04:41 1m 28s
nsys-jax: compatibility with Nsight Systems 2024.5 nsys-jax pure-Python CI #141: Pull request #985 opened by olupton
August 8, 2024 14:50 1m 38s olupton/nsys-2024.5
olupton/nsys-2024.5
August 8, 2024 14:50 1m 38s
fix remat policy and fused attn variable in test-maxtext.sh nsys-jax pure-Python CI #140: Pull request #982 opened by kocchop
August 6, 2024 23:16 1m 48s faysal/maxtext-test-script
faysal/maxtext-test-script
August 6, 2024 23:16 1m 48s
test-maxtext.sh: set default mem fraction to 0.9 (#979) nsys-jax pure-Python CI #139: Commit 4696e4d pushed by kocchop
August 6, 2024 21:22 1m 24s main
main
August 6, 2024 21:22 1m 24s
Adds patch to rosetta-pax for PP fix nsys-jax pure-Python CI #138: Pull request #981 opened by terrykong
August 6, 2024 20:41 1m 40s add-kaixi-pp-patch
add-kaixi-pp-patch
August 6, 2024 20:41 1m 40s
test-maxtext.sh: set default mem fraction to 0.9 nsys-jax pure-Python CI #137: Pull request #979 synchronize by nouiz
August 6, 2024 13:29 1m 43s sergachev:patch-1
sergachev:patch-1
August 6, 2024 13:29 1m 43s
test-maxtext.sh: set default mem fraction to 0.9 nsys-jax pure-Python CI #136: Pull request #979 synchronize by sergachev
August 5, 2024 21:31 1m 45s sergachev:patch-1
sergachev:patch-1
August 5, 2024 21:31 1m 45s
test-maxtext.sh: set default mem fraction to 0.9 nsys-jax pure-Python CI #135: Pull request #979 opened by sergachev
August 5, 2024 21:13 1m 51s sergachev:patch-1
sergachev:patch-1
August 5, 2024 21:13 1m 51s
Added model support to test-maxtext.sh (#971) nsys-jax pure-Python CI #134: Commit 3f6999e pushed by nouiz
August 3, 2024 01:27 2m 6s main
main
August 3, 2024 01:27 2m 6s
Fixes Imagen sampling example and updates container nsys-jax pure-Python CI #133: Pull request #868 synchronize by terrykong
August 2, 2024 23:51 1m 29s imagen-fix
imagen-fix
August 2, 2024 23:51 1m 29s
Added model support to test-maxtext.sh nsys-jax pure-Python CI #132: Pull request #971 synchronize by nouiz
August 2, 2024 19:23 1m 36s faysal/maxtext-test-script
faysal/maxtext-test-script
August 2, 2024 19:23 1m 36s
Remove grok nsys-jax pure-Python CI #131: Pull request #975 opened by DwarKapex
August 2, 2024 18:33 1m 32s vkozlov/remove-grok
vkozlov/remove-grok
August 2, 2024 18:33 1m 32s
Added model support to test-maxtext.sh nsys-jax pure-Python CI #130: Pull request #971 synchronize by kocchop
August 1, 2024 21:06 2m 24s faysal/maxtext-test-script
faysal/maxtext-test-script
August 1, 2024 21:06 2m 24s
Added model support to test-maxtext.sh nsys-jax pure-Python CI #129: Pull request #971 opened by kocchop
August 1, 2024 01:31 1m 28s faysal/maxtext-test-script
faysal/maxtext-test-script
August 1, 2024 01:31 1m 28s
Update CUDA to public 12.5.1 nsys-jax pure-Python CI #128: Pull request #968 opened by DwarKapex
July 30, 2024 00:02 1m 27s cuda-12.5.1-base-image
cuda-12.5.1-base-image
July 30, 2024 00:02 1m 27s