Skip to content

Actions: NVIDIA/JAX-Toolbox

Assistant

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
855 workflow runs
855 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Bump CUDA to 12.5
Assistant #831: Issue comment #897 (comment) created by terrykong
July 17, 2024 16:14 12s
July 17, 2024 16:14 12s
Temp WAR for ssh to Slurm Cluster
Assistant #830: Issue comment #947 (comment) created by nouiz
July 16, 2024 17:51 9s
July 16, 2024 17:51 9s
nsys-jax: example of inspecting heap traces in compile-time metadata
Assistant #829: Issue comment #902 (comment) created by olupton
July 3, 2024 14:15 14s
July 3, 2024 14:15 14s
Add maxtext for arm64 dockerfile.
Assistant #828: Issue comment #921 (comment) created by gpupuck
June 26, 2024 22:24 12s
June 26, 2024 22:24 12s
Update DALI pipeline for ViT
Assistant #827: Issue comment #911 (comment) created by terrykong
June 26, 2024 06:48 19s
June 26, 2024 06:48 19s
Add maxtext for arm64 dockerfile.
Assistant #826: Issue comment #914 (comment) created by nouiz
June 25, 2024 12:55 13s
June 25, 2024 12:55 13s
Add maxtext for arm64 dockerfile.
Assistant #825: Issue comment #914 (comment) created by olupton
June 24, 2024 14:42 12s
June 24, 2024 14:42 12s
Ping numpy to 1.24.3
Assistant #824: Issue comment #908 (comment) created by nouiz
June 19, 2024 16:54 12s
June 19, 2024 16:54 12s
Ping numpy to 1.24.3
Assistant #823: Issue comment #908 (comment) created by olupton
June 19, 2024 07:47 11s
June 19, 2024 07:47 11s
nsys-jax: example of inspecting heap traces in compile-time metadata
Assistant #822: Issue comment #902 (comment) created by nouiz
June 18, 2024 13:25 10s
June 18, 2024 13:25 10s
Fix Triton build again
Assistant #821: Issue comment #899 (comment) created by olupton
June 14, 2024 07:24 11s
June 14, 2024 07:24 11s
WAR to build upstream-pax for ARM64
Assistant #820: Issue comment #898 (comment) created by DwarKapex
June 13, 2024 19:00 13s
June 13, 2024 19:00 13s
Fix Triton build again
Assistant #819: Issue comment #899 (comment) created by DwarKapex
June 13, 2024 17:05 13s
June 13, 2024 17:05 13s
Add FMHA PAXML test
Assistant #818: Issue comment #830 (comment) created by ashors1
June 12, 2024 05:48 20s
June 12, 2024 05:48 20s
Add FMHA PAXML test
Assistant #817: Issue comment #830 (comment) created by hmonishN
June 11, 2024 17:01 10s
June 11, 2024 17:01 10s
Add FMHA PAXML test
Assistant #816: Issue comment #830 (comment) created by DwarKapex
June 10, 2024 22:46 12s
June 10, 2024 22:46 12s
Port 24.04 changes into main
Assistant #815: Issue comment #851 (comment) created by terrykong
June 7, 2024 16:29 11s
June 7, 2024 16:29 11s
Add Grok patches to paxml and praxis patchlists
Assistant #814: Issue comment #884 (comment) created by ashors1
June 5, 2024 22:55 13s
June 5, 2024 22:55 13s
ci: Selective triggering
Assistant #813: Issue comment #775 (comment) created by DwarKapex
June 5, 2024 17:53 9s
June 5, 2024 17:53 9s
Reduce number of t5x tests
Assistant #812: Issue comment #880 (comment) created by hmonishN
June 5, 2024 00:58 12s
June 5, 2024 00:58 12s
nsys-jax: add basic CI, support all-to-all and repeated thunks
Assistant #811: Issue comment #875 (comment) created by olupton
June 4, 2024 11:45 10s
June 4, 2024 11:45 10s
[NVIDIA] Update XLA-fp8 docs for ninja user guide
Assistant #810: Issue comment #478 (comment) created by kaixih
May 30, 2024 19:38 13s
May 30, 2024 19:38 13s
nsys-jax: argparse+docs, fix post-processing, analyse collectives
Assistant #809: Issue comment #863 (comment) created by olupton
May 30, 2024 09:21 11s
May 30, 2024 09:21 11s
CI: Prevent metrics step from failing when tensorboard dir is not found
Assistant #808: Issue comment #862 (comment) created by ashors1
May 29, 2024 17:27 15s
May 29, 2024 17:27 15s
WAR for JAX unit-tests
Assistant #807: Issue comment #832 (comment) created by terrykong
May 29, 2024 15:46 11s
May 29, 2024 15:46 11s