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

80 Open 102 Closed
80 Open 102 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
Flag xla_gpu_enable_address_computation_fusion was renamed to xla_gpu_enable_dynamic_slice_fusion
#1098 opened Oct 13, 2024 by qGentry
Register-level FLOPS counting in JAX programs ?
#836 opened May 18, 2024 by mitkotak
@olupton
2
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
Report SLURM runner status as launch-slurm-runner job outcome
#540 opened Feb 14, 2024 by yhtang
@DwarKapex
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
1
[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
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
Use sbatch-action for more robust SLURM job handling
#407 opened Dec 4, 2023 by yhtang
VCS pinning in pip-compile within pip-finalize.sh
#384 opened Nov 17, 2023 by terrykong
@yhtang @DwarKapex
Split nightly-rosetta-*-build-test.yaml into build and test files
#376 opened Nov 16, 2023 by DwarKapex
1
Make the CI stable, remove noisy random failures.
#341 opened Oct 25, 2023 by yhtang
Finalize/documet the "stable" tag
#340 opened Oct 25, 2023 by yhtang
JET nightly
#336 opened Oct 25, 2023 by yhtang
Fix the nightly CI tests
#335 opened Oct 25, 2023 by yhtang
3 of 11 tasks
[Bot] upstream-pax test failures on 2023-10-16 auto-triage Issues that were tagged by GH Actions
#318 opened Oct 18, 2023 by github-actions bot
@ashors1
Remove --xla_gpu_enable_triton_gemm false
#317 opened Oct 17, 2023 by nouiz
2
Publish single-arch images alongside multi-arch images
#311 opened Oct 16, 2023 by yhtang
OCI image types not supported on clusters with enroot < 3.4.1
#310 opened Oct 16, 2023 by terrykong
@yhtang
5
ProTip! Type g i on any issue or pull request to go back to the issue listing page.