Skip to content

Issues: jax-ml/jax

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

Clear current search query, filters, and sorts
742 Open 2,407 Closed
742 Open 2,407 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Support mask gather in Mosaic compiler bug Something isn't working
#29092 opened May 29, 2025 by xy12181
@justinjfu
2
It works fine on a single GPU, but does not function properly in a multi-GPU (parallel GPU) setup. bug Something isn't working
#29088 opened May 29, 2025 by juju0111
The contents of the .pyi file and the docstring do not correspond to reality. bug Something isn't working
#29083 opened May 29, 2025 by DanisNone
@jakevdp
Runtime failure/regression while running inference on JAX 0.6.1 bug Something isn't working
#29070 opened May 28, 2025 by Rohanjames1997
1
An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu. bug Something isn't working
#29068 opened May 28, 2025 by galah92
1
lax.cond crashes on Windows bug Something isn't working
#29049 opened May 28, 2025 by lnabergall
PyTreeDef comparison fails type checking for jax==0.6.1 bug Something isn't working
#29037 opened May 27, 2025 by mfschubert
@superbobry
3
jax.numpy.dot fails with an internal error bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#29031 opened May 27, 2025 by robagar
2
pallas compile fails with bfloat16 but works with float32 bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#29002 opened May 25, 2025 by eamartin
@sharadmv
4
AttributeError: partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import) bug Something isn't working
#29000 opened May 25, 2025 by mvds314
4
jax.numpy.ndarray.at OOB indexing with negative indices and mode="fill" not coorect bug Something isn't working
#28998 opened May 24, 2025 by geligeli
@jakevdp
1
build on Jetson AGX Orin failed bug Something isn't working
#28989 opened May 23, 2025 by wjz-kidding
RuntimeError: Unable to initialize backend 'cuda': FAILED_PRECONDITION: No visible GPU devices. bug Something isn't working
#28980 opened May 23, 2025 by EKHess
jax.nn.dot_product_attention with implementation='cudnn' fails when only mask, but no bias provided bug Something isn't working
#28974 opened May 23, 2025 by danjenson
3
JAX/FALX.NNX: TypeError - JVPTracer for scalar nnx.Param incorrectly treated as compile-time constant in nested transformations (jit + value_and_grad) bug Something isn't working
#28937 opened May 22, 2025 by Beiglar
2
NotImplementedError: MLIR translation rule for primitive 'schur' not found for platform cuda bug Something isn't working
#28927 opened May 22, 2025 by YousefElbrolosy
@dfm
3
Segfault in simple (1-line) deserialized AOT-compilation workload bug Something isn't working
#28908 opened May 21, 2025 by joaospinto
NotImplementedError: MLIR translation rule for primitive 'debug_callback' not found for platform METAL Apple GPU (Metal) plugin bug Something isn't working
#28786 opened May 16, 2025 by amavrits
@kulinseth @shuhand0
3
debug.print inside scan gets lost during AD bug Something isn't working
#28738 opened May 14, 2025 by dfm
@dfm
5
CUDA_ERROR_ILLEGAL_ADDRESS thrown by XlaRuntimeError bug Something isn't working
#28710 opened May 13, 2025 by DerPlayer2001
6
segmentation fault (core dumped) while using gpu bug Something isn't working
#28692 opened May 12, 2025 by xywawawa
@rdyro
6
Nondeterministic behavior for pytree extensions if aux data does not have equality bug Something isn't working
#28659 opened May 9, 2025 by batterseapower
@jakevdp
6
0.6 Installation issues and 0.5 segfaults bug Something isn't working
#28623 opened May 8, 2025 by GJBoth
@hawkinsp
5
jaxlib doesn't have a pyproject.toml bug Something isn't working
#28582 opened May 7, 2025 by olupton
1
Possible data race in PyOperation and ~PyOperation on cached Module bug Something isn't working free threading Issues found in free threading builds
#28551 opened May 6, 2025 by vfdev-5
1
2
ProTip! What’s not been updated in a month: updated:<2025-04-30.