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

1,508 Open 4,509 Closed
1,508 Open 4,509 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

expi function causes tracer leak bug Something isn't working
#26972 opened Mar 6, 2025 by fehiepsi
1
@jakevdp
Pallas TPU kernel: VerificationError when trying to do async copy from Refs at dynamic offsets bug Something isn't working
#26971 opened Mar 6, 2025 by wilson-nomic
jax-metal: Installation according to apples guide does not work. bug Something isn't working
#26968 opened Mar 6, 2025 by PhilippFoerster
@kulinseth @shuhand0
[mutable_array] Indexing a ref using x_ref[...] under jit does not preserve its sharding spec bug Something isn't working
#26936 opened Mar 5, 2025 by ayaka14732
@mattjj @dougalm
1
Failed build: CI - with Numpy/Scipy nightly wheels (nightly) build failed
#26933 opened Mar 5, 2025 by github-actions bot
2
pyright: pmap type hint missing bug Something isn't working
#26921 opened Mar 4, 2025 by lockwo
4
Bwd pass in ragged_all_to_all is not supported bug Something isn't working
#26915 opened Mar 4, 2025 by RissyRan
@mattjj
4
Make JAX export fail if the mesh shape on save is different than the mesh shape on load bug Something isn't working
#26914 opened Mar 4, 2025 by bartchr808
"select cases must have the same shapes" using checkify and shard_map bug Something isn't working
#26890 opened Mar 3, 2025 by Findus23
@yashk2810
1
checkify breaks inspect_array_sharding bug Something isn't working
#26889 opened Mar 3, 2025 by Findus23
"Primitive gt requires argument replication types to match" when using checkify with shard_map bug Something isn't working
#26887 opened Mar 3, 2025 by Findus23
Expose get_max_identity and get_min_identity to the public API enhancement New feature or request
#26877 opened Mar 1, 2025 by carlosgmartin
@jakevdp
1
Results mismatch between different GEMM algorithms on AMD Instinct accelerator bug Something isn't working
#26846 opened Feb 28, 2025 by ffrancesco94
[Jax GPU] Unexpected difference when results should be equal bug Something isn't working
#26836 opened Feb 28, 2025 by hanzhi713
Fix tokens for arrays with no local shards bug Something isn't working
#26815 opened Feb 27, 2025 by emilyfertig
3
[Pallas] python for loop in kernel function is unbearably slow compared to fori_loop bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#26812 opened Feb 27, 2025 by ysngshn
5
expm yields wrong values even with max_squarings set to high values bug Something isn't working
#26802 opened Feb 27, 2025 by ymardoukhi
@jakevdp
1
Implement custom_linear_solve and eigh primtive rules for Taylor series propagation in jax.experimental.jet enhancement New feature or request
#26801 opened Feb 27, 2025 by yachmena
1
Improve debug_nan in shard_map to pinpoint exact line of NaN occurrence enhancement New feature or request
#26796 opened Feb 27, 2025 by Stella-S-Yan
Jax compilation hangs indefinitely bug Something isn't working
#26767 opened Feb 26, 2025 by n-gao
__array_priority__ is not checked by JAX for downstream custom arrays bug Something isn't working
#26764 opened Feb 26, 2025 by gautierronan
@jakevdp
6
shard_map not working with symbolic_zero in custom_jvp enhancement New feature or request
#26763 opened Feb 26, 2025 by n-gao
@mattjj
1
Termination timeout error bug Something isn't working
#26758 opened Feb 26, 2025 by andrewilyas
1
AttributeError: module 'jaxlib.xla_client' has no attribute 'register_custom_type_id_handler'. bug Something isn't working
#26750 opened Feb 25, 2025 by rohandekate10
@justinjfu
6
Matrix multiply incorrect for large tensors bug Something isn't working
#26689 opened Feb 23, 2025 by unalmis
1
ProTip! What’s not been updated in a month: updated:<2025-02-06.