Skip to content

Issues: NVIDIA/JAX-Toolbox

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

116 Open 66 Closed
116 Open 66 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
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Feature reality models similar to exist LLM, like medical target expectations.
#1202 opened Dec 16, 2024 by linjing-lab
xla_gpu_pgle_accuracy_checker is set to WARN, but no profile path specified in xla_gpu_pgle_profile_file_or_directory_path
#1135 opened Oct 31, 2024 by andportnoy
1
Flag xla_gpu_enable_address_computation_fusion was renamed to xla_gpu_enable_dynamic_slice_fusion
#1098 opened Oct 13, 2024 by qGentry
Need to implement a bastion runner for submitting CI jobs.
#1093 opened Oct 10, 2024 by yhtang
@DwarKapex
CUDA Forward Compatibility mode
#1022 opened Aug 29, 2024 by luyug
1
Dockerfile for images
#973 opened Aug 1, 2024 by shivajid
1
'jax.experimental.maps' import error
#962 opened Jul 25, 2024 by MikeMpapa
2
Register-level FLOPS counting in JAX programs ?
#836 opened May 18, 2024 by mitkotak
@olupton
2
Verify if the receive datapath manager is already enabled in DGXcloud on GCP
#722 opened Apr 16, 2024 by yhtang
Provide versioned tags instead of date tags
#717 opened Apr 13, 2024 by samos123
Multi-cluster support in CI
#616 opened Mar 5, 2024 by yhtang
T5X issue tracker - pw_checkpoints undefined
#610 opened Mar 4, 2024 by yhtang
Fix Levanter test
#596 opened Feb 28, 2024 by DwarKapex
Install the latest LLVM
#580 opened Feb 22, 2024 by yhtang
Use the unit testing framework instead of the MGMN testing framework for _test_te.yaml
#560 opened Feb 16, 2024 by yhtang
@terrykong @ashors1
Report SLURM runner status as launch-slurm-runner job outcome
#540 opened Feb 14, 2024 by yhtang
@DwarKapex
Let T5X and Paxml arm64 build share the tftext build stage
#531 opened Feb 12, 2024 by yhtang
Unique name and logfile for SLURM-launched A100 runners
#525 opened Feb 8, 2024 by yhtang
Implement MFU and peak memory usage metric collection.
#519 opened Feb 6, 2024 by hmonishN
Add a test to check that NVCC is used for CUDA and Clang is used for pure host compilation
#515 opened Feb 5, 2024 by andportnoy
[Feature Request] Need Matmul Attention layer instead of Einsum to support GPU running
#502 opened Jan 27, 2024 by MoFHeka
4
Warn users not to run scripts off /opt?
#475 opened Jan 16, 2024 by yhtang
Add stable tag to all upstream containers
#474 opened Jan 16, 2024 by yhtang
Pallas build duplicates work bug a bug of JAX-Toolbox itself instead of the JAX ecosystem software that it ships good first issue Good for newcomers
#467 opened Jan 11, 2024 by olupton
2
Need to reduce the number of multi-node jobs per workflow run
#432 opened Dec 18, 2023 by yhtang
ProTip! no:milestone will show everything without a milestone.