Skip to content

Pull requests: NVIDIA/JAX-Toolbox

New pull request New
44 Open 1,017 Closed
44 Open 1,017 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Replace deprecated flag xla_gpu_graph_level.
#1244 opened Jan 11, 2025 by sergachev Loading…
[nsys-jax] Add ratio of hidden communication time to total communication time
#1241 opened Jan 10, 2025 by sfvaroglu Loading…
1
Remove deprecated flag xla_gpu_enable_triton_softmax_fusion.
#1240 opened Jan 10, 2025 by sergachev Loading…
Remove V100 from test environment
#1238 opened Jan 9, 2025 by DwarKapex Loading…
CI: run MaxText tests on AWS with NGC release candidate images
#1237 opened Jan 9, 2025 by olupton Loading…
6
CI: run NCCL tests on AWS with NGC release candidate images
#1234 opened Jan 9, 2025 by olupton Loading…
Move nccl-tests-on-k8s logic into a reusable workflow
#1233 opened Jan 9, 2025 by olupton Loading…
nsys-jax: optimise data loading and .zip creation
#1193 opened Dec 10, 2024 by olupton Loading…
5
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…
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 ray to the base image
#1067 opened Sep 27, 2024 by yhtang Loading…
@yhtang
3
Add JetStream to MaxText container
#1058 opened Sep 23, 2024 by DwarKapex Loading…
@DwarKapex
8
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
Enable TE fused attention by default in rosetta pax container
#757 opened Apr 23, 2024 by ashors1 Loading…
Experiment with IB/OFED libs from distro repo
#756 opened Apr 23, 2024 by yhtang Loading…
ProTip! Type g i on any issue or pull request to go back to the issue listing page.