Skip to content

Pull requests: NVIDIA/JAX-Toolbox

New pull request New
46 Open 2,101 Closed
46 Open 2,101 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

Revert "fix versions for tf and tf-text to prevent bug (#2287)"
#2331 opened Sep 17, 2026 by aybchan Member Loading…
3
Rename -scale-training to -staging
#2324 opened Sep 11, 2026 by olupton Collaborator Loading…
3
docs: add DeepSeek-v3 MoE reference configuration for MaxText
#2322 opened Sep 10, 2026 by shljessie Contributor Loading…
2 of 3 tasks
1
alphafold3: bump tokamax pin
#2321 opened Sep 10, 2026 by olupton Collaborator Loading…
3
Update 26.09-devel from main
#2315 opened Sep 8, 2026 by aybchan Member Loading…
1
--commit feature to triage tool
#2298 opened Aug 28, 2026 by Steboss Contributor Loading…
7
Add jax-cudnn-frontend agent skill
#2273 opened Aug 17, 2026 by jessicadesilva Member Loading…
8
Bump js-yaml from 4.2.0 to 4.3.1
#2263 opened Aug 10, 2026 by katsugtgz Loading…
1
Remove staging image docs
#2252 opened Aug 4, 2026 by aybchan Member Loading…
2
Improve build-te for build-jax step in CI
#2245 opened Jul 30, 2026 by Steboss Contributor • Draft
21
Document scheduler memory fencing flags
#2242 opened Jul 29, 2026 by shawnwang18 Loading…
1
General cleaning and fixing of ci
#2213 opened Jul 9, 2026 by Steboss Contributor Loading…
5
Set to load-balance jobs on GKE
#2151 opened Jun 5, 2026 by aybchan Member Loading…
3
Test new tf on axlearn for EFA
#2116 opened May 20, 2026 by Steboss Contributor • Draft
29
Refactor of vLLM bridge
#1914 opened Jan 23, 2026 by keshavb96 Contributor Loading…
7
Install uv (https://docs.astral.sh/uv/) in base container
#1891 opened Jan 12, 2026 by andportnoy Contributor Loading…
[jax-inference-offloading] consolidate definitions for default tensor dtype
#1816 opened Dec 5, 2025 by yhtang Contributor Loading…
8
[jax-inference-offloading] Bump to CUDA 13.0
#1807 opened Dec 2, 2025 by yhtang Contributor • Draft
Add MACE training example
#1192 opened Dec 10, 2024 by mariogeiger Loading…
Customize NCCL for base container
#1123 opened Oct 28, 2024 by DwarKapex Contributor Loading…
2
ci[maxtext]: adds rosetta directory to maxtext containers
#1097 opened Oct 13, 2024 by terrykong Contributor Loading…
2
chore: remove deprecated flags from model xla flag configs
#1088 opened Oct 8, 2024 by terrykong Contributor Loading…
1
Fix the Order of GatedAct to be (act, linear) when Pax2TE.
#892 opened Jun 10, 2024 by mingxu1067 Contributor Loading…
Add upstream pax 2-node Grok test
#854 opened May 24, 2024 by ashors1 Contributor Loading…
2
Support auto-generated FP8 meta for CKPT converters.
#844 opened May 21, 2024 by mingxu1067 Contributor Loading…
4
ProTip! Adding no:label will show everything without a label.