Skip to content

Pull requests: NVIDIA/JAX-Toolbox

New pull request New
38 Open 638 Closed
38 Open 638 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Add gemma to tag collector for mightly publishing
#849 opened May 23, 2024 by DwarKapex Loading…
@DwarKapex
Support auto-generated FP8 meta for CKPT converters.
#844 opened May 21, 2024 by mingxu1067 Loading…
Revert "Adds CUDA_MODULE_LOADING=EAGER to core jax container env vars"
#842 opened May 21, 2024 by olupton Loading…
Update README.md - PaliGemma
#839 opened May 20, 2024 by sbhavani Loading…
WAR for JAX unit-tests
#832 opened May 17, 2024 by DwarKapex Loading…
@DwarKapex
4
Add FMHA PAXML test
#830 opened May 16, 2024 by hmonishN Loading…
18
nsys-jax: handle parallel compilation
#822 opened May 14, 2024 by olupton Loading…
5
Run fewer levanter tests
#811 opened May 8, 2024 by DwarKapex • Draft
@DwarKapex
Add automatic triage utility
#793 opened May 2, 2024 by olupton Loading…
1
Use HEAD versions in presubmit CI
#782 opened May 1, 2024 by olupton Loading…
14
ci: Selective triggering
#775 opened Apr 30, 2024 by ko3n1g Loading…
1
10
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…
Bump CUDA to 12.4 + bump OFED version to 24.01
#755 opened Apr 23, 2024 by yhtang Loading…
Check compute capability at runtime
#751 opened Apr 22, 2024 by DwarKapex Loading…
@DwarKapex
32
Install ray on all images
#750 opened Apr 22, 2024 by chaserileyroberts Loading…
1
ci: Restore rosetta-t5x-test
#749 opened Apr 22, 2024 by ko3n1g Loading…
4
feat: Add rosetta-maxtext
#738 opened Apr 18, 2024 by ko3n1g Loading…
16
Update front-page readme with link to XLA flag doc
#684 opened Apr 4, 2024 by terrykong Loading…
9
Update NCCL version on demand
#658 opened Mar 22, 2024 by DwarKapex Loading…
@DwarKapex
14
Re-enable environment variables to guard against hangs with Triton gemms
#624 opened Mar 7, 2024 by ashors1 • Draft
6
Address issue #540
#554 opened Feb 15, 2024 by DwarKapex • Draft
@DwarKapex
7
Use Clang to build MJX
#534 opened Feb 13, 2024 by DwarKapex • Draft
@DwarKapex
3
Install TRT-LLM/TRT OSS
#510 opened Feb 1, 2024 by yhtang • Draft
@yhtang
Pins jax==0.4.23 (xla matched), flax==0.7.5, te==1.2.1
#493 opened Jan 23, 2024 by terrykong • Draft
2
ProTip! Updated in the last three days: updated:>2024-05-20.