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

Allow multi-GPU tests and enable MPS
#1153 opened Nov 13, 2024 by olupton Loading…
CI: example job using EKS jumphost
#1151 opened Nov 12, 2024 by olupton Loading…
Update user buffers documentation
#1144 opened Nov 6, 2024 by trevor-m 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…
Levanter: Resolve issue with zarr3
#1063 opened Sep 25, 2024 by DwarKapex 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…
Install ray on all images
#750 opened Apr 22, 2024 by chaserileyroberts 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! Add no:assignee to see everything that’s not assigned.