Skip to content

Pull requests: NVIDIA/JAX-Toolbox

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

fix t5x tests
#1208 opened Dec 19, 2024 by ashors1 Loading…
nsys-jax: optimise data loading and .zip creation
#1193 opened Dec 10, 2024 by olupton Loading…
Add MACE training example
#1192 opened Dec 10, 2024 by mariogeiger Loading…
Allow nsys 2024.7 installation.
#1176 opened Nov 28, 2024 by olupton Loading…
Customize NCCL for base container
#1123 opened Oct 28, 2024 by DwarKapex Loading…
Add cuda profiler hook to paxml
#1089 opened Oct 9, 2024 by shawnwang18 Loading…
Add ray to the base image
#1067 opened Sep 27, 2024 by yhtang Loading…
Add JetStream to MaxText container
#1058 opened Sep 23, 2024 by DwarKapex Loading…
Measure final docker image size
#893 opened Jun 10, 2024 by DwarKapex Draft
Add upstream pax 2-node Grok test
#854 opened May 24, 2024 by ashors1 Loading…
Add FMHA PAXML test
#830 opened May 16, 2024 by hmonishN Loading…
ci: Selective triggering
#775 opened Apr 30, 2024 by ko3n1g Loading…
Experiment with IB/OFED libs from distro repo
#756 opened Apr 23, 2024 by yhtang Loading…
ci: Restore rosetta-t5x-test
#749 opened Apr 22, 2024 by ko3n1g Loading…
feat: Add rosetta-maxtext
#738 opened Apr 18, 2024 by ko3n1g Loading…
Address issue #540
#554 opened Feb 15, 2024 by DwarKapex Draft
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.