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,512 Open 4,522 Closed
1,512 Open 4,522 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

JAX build fails when using clang 19 bug Something isn't working
#27091 opened Mar 12, 2025 by pearu
TSAN race in lax_numpy_operators_test.py under free-threading bug Something isn't working free threading Issues found in free threading builds
#27089 opened Mar 12, 2025 by vfdev-5
1
State of jax.scipy.special functions: tested by evaluation or autograd, incorrectness and missing functionality
#27088 opened Mar 12, 2025 by pearu
Jax Tensorboard profiling example not working bug Something isn't working
#27070 opened Mar 11, 2025 by Teculos
XlaRuntimeError on latest jax-metal Apple GPU (Metal) plugin bug Something isn't working
#27062 opened Mar 11, 2025 by limarta
@kulinseth @shuhand0
DotGeneral for >2D Tensors in Pallas on TPU enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#27061 opened Mar 11, 2025 by jondeaton
@sharadmv
Not implemented: Lane broadcast when trying to use reductions bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#27051 opened Mar 11, 2025 by wilson-nomic
1
In a Kubernetes (K8s) environment, the number of threads during JAX initialization is determined based on the CPU of the node where the pod resides, rather than the CPU resources requested by the pod. bug Something isn't working
#27028 opened Mar 10, 2025 by mochen302
Comparison of large unsigned and signed integers returns incorrect results bug Something isn't working
#27021 opened Mar 9, 2025 by pearu
@jakevdp
5
Improve Performance jnp.argsort enhancement New feature or request
#27004 opened Mar 7, 2025 by cdelv
2
jax.random depends on precision bug Something isn't working
#27002 opened Mar 7, 2025 by xaviergonzalez
@jakevdp
2
jax.debug.print Displays Misleading Output for XLA 4-bit Tensors bug Something isn't working
#26998 opened Mar 7, 2025 by wenscarl
@danielsuo
3
jax-metal: Installation according to apples guide does not work. bug Something isn't working
#26968 opened Mar 6, 2025 by PhilippFoerster
@kulinseth @shuhand0
Failed build: CI - with Numpy/Scipy nightly wheels (nightly) build failed
#26933 opened Mar 5, 2025 by github-actions bot
1
@pearu
10
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
ProTip! Exclude everything labeled bug with -label:bug.