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

111 Open 61 Closed
111 Open 61 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
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

Register-level FLOPS counting in JAX programs ?
#836 opened May 18, 2024 by mitkotak
Update install-cudnn.sh script
#824 opened May 14, 2024 by DwarKapex
@DwarKapex
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
Unify presubmit CI and nightly CI
#416 opened Dec 7, 2023 by terrykong
@DwarKapex
1
Environment data to be dumped for performance benchmarking.
#414 opened Dec 6, 2023 by hmonishN
Providing containers with newer Python versions(?) enhancement New feature or request question Further information is requested
#410 opened Dec 5, 2023 by olupton
1
Move lingvo patches into the manifest
#409 opened Dec 4, 2023 by terrykong
build-jax.sh Should have an option --bazel_only that only install bazel.
#408 opened Dec 4, 2023 by nouiz
Use sbatch-action for more robust SLURM job handling
#407 opened Dec 4, 2023 by yhtang
ProTip! Adding no:label will show everything without a label.