Skip to content

Pull requests: NVIDIA/JAX-Toolbox

New pull request New
45 Open 1,478 Closed
45 Open 1,478 Closed
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
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Remove NCCL_NVLS_ENABLE setting from JAX image
#1708 opened Oct 3, 2025 by aybchan Loading…
1 of 2 tasks
Remove NCCL_NVLS_ENABLE setting from JAX image
#1707 opened Oct 3, 2025 by aybchan Loading…
25.10 NGC release testing
#1693 opened Sep 26, 2025 by aybchan Loading…
1
Add NGC workflow
#1692 opened Sep 26, 2025 by aybchan Loading…
1 task done
@aybchan
[WIP] Jax Cutlass DSL Support Updates for Cutlass 4.2
#1680 opened Sep 19, 2025 by mgoldfarb-nvidia Loading…
Add optimization level and lookup sections
#1678 opened Sep 18, 2025 by terryysun Loading…
14
Test NCCL with EKS
#1651 opened Sep 1, 2025 by Steboss • Draft
Test on the new AXLearn version
#1595 opened Aug 1, 2025 by Steboss • Draft
5
Add basic sitrep steps for GKE XPK action
#1579 opened Jul 25, 2025 by aybchan Loading…
@aybchan
25.08: Add NGC release testing
#1554 opened Jul 15, 2025 by aybchan Loading…
Add MACE example from Mario Geiger
#1341 opened Mar 12, 2025 by DwarKapex Loading…
@DwarKapex
1
Simplify the way we're generating badges
#1278 opened Jan 30, 2025 by Steboss Loading…
2
Add MACE training example
#1192 opened Dec 10, 2024 by mariogeiger Loading…
Customize NCCL for base container
#1123 opened Oct 28, 2024 by DwarKapex Loading…
2
ci[maxtext]: adds rosetta directory to maxtext containers
#1097 opened Oct 13, 2024 by terrykong Loading…
2
Add cuda profiler hook to paxml
#1089 opened Oct 9, 2024 by shawnwang18 Loading…
6
chore: remove deprecated flags from model xla flag configs
#1088 opened Oct 8, 2024 by terrykong Loading…
1
Add missing port for ssh connection for upstream T5x test
#1075 opened Oct 2, 2024 by DwarKapex Loading…
3
Add JetStream to MaxText container
#1058 opened Sep 23, 2024 by DwarKapex Loading…
@DwarKapex
14
Measure final docker image size
#893 opened Jun 10, 2024 by DwarKapex • Draft
Fix the Order of GatedAct to be (act, linear) when Pax2TE.
#892 opened Jun 10, 2024 by mingxu1067 Loading…
Add upstream pax 2-node Grok test
#854 opened May 24, 2024 by ashors1 Loading…
2
Support auto-generated FP8 meta for CKPT converters.
#844 opened May 21, 2024 by mingxu1067 Loading…
4
Add FMHA PAXML test
#830 opened May 16, 2024 by hmonishN Loading…
33
ci: Selective triggering
#775 opened Apr 30, 2024 by ko3n1g Loading…
1
28
ProTip! What’s not been updated in a month: updated:<2025-09-03.