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,528 Open 4,537 Closed
1,528 Open 4,537 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

Error message from CUDA backend compiler for some JAX operations bug Something isn't working
#27335 opened Mar 21, 2025 by kcdodd
scatter_determinism_expander is incompatible with 64-bit precision bug Something isn't working
#27324 opened Mar 21, 2025 by wrvsrx
Spurious leak checker error when using non-local variable across jitted functions bug Something isn't working
#27315 opened Mar 21, 2025 by ayaka14732
@dougalm
1
object has no attribute bug Something isn't working
#27314 opened Mar 21, 2025 by SandSnip3r
1
Using jax.Device as a dict key for pytree transformations enhancement New feature or request
#27300 opened Mar 20, 2025 by shoyer
6
Gemma 3 + jax-metal: 'mhlo.convolution' op Not supported Apple GPU (Metal) plugin bug Something isn't working
#27288 opened Mar 20, 2025 by FL33TW00D
@kulinseth @shuhand0
1
Unsafe cast u32 -> i32 in threefry_seed question Questions for the JAX team
#27282 opened Mar 20, 2025 by botev-openai
@jakevdp
3
Document PGLE in JAX documentation enhancement New feature or request
#27269 opened Mar 20, 2025 by jprabhas
1
Feature request: linalg.eig using nvidia cusolver enhancement New feature or request
#27265 opened Mar 19, 2025 by mfschubert
@dfm
4
TSAN race in //tests:dtypes_test_cpu under free-threading bug Something isn't working free threading Issues found in free threading builds
#27244 opened Mar 19, 2025 by vfdev-5
1
Segfault during scan function (jax 5.2, nvidia -drivers 570, cuda 12.8) bug Something isn't working
#27235 opened Mar 18, 2025 by alik-git
Choosing CPU backend causes JAX to allocate GPU memory bug Something isn't working
#27232 opened Mar 18, 2025 by cool-RR
4
Stacking flax.nnx.Module layers with vmap and forwarding with scan result in loss of precision in XLA backend bug Something isn't working
#27228 opened Mar 18, 2025 by HeavyCrab
10
Pallas GPU FlashAttention forward only works when sequence lengths are multiples of block size bug Something isn't working
#27224 opened Mar 18, 2025 by yuzu-ido
XlaRuntimeError error when calling jax.lax.with_sharding_constraint with an Array of PRNG keys inside a jited function bug Something isn't working
#27223 opened Mar 18, 2025 by vadmbertr
Drop in speed on GPU from version 0.4.35 to 0.4.36 bug Something isn't working
#27218 opened Mar 18, 2025 by patrick-egenlauf
Segfault on Linux with jvp-of-jvp-of-traced-multiplication bug Something isn't working
#27201 opened Mar 17, 2025 by johannahaffner
3
Export jax._src.typing.Shape to jax.typing. enhancement New feature or request
#27200 opened Mar 17, 2025 by ZacCranko
1
@jakevdp
3
ptxas error: Fusion uses too much shared data bug Something isn't working
#27190 opened Mar 17, 2025 by emergenz
1
Results do not match the reference. This is likely a bug/unexpected loss of precision. bug Something isn't working
#27188 opened Mar 17, 2025 by AaronSpieler
1
JAX shows more ROCM devices than GPUs after calling jax.distributed.initialize() bug Something isn't working
#27187 opened Mar 17, 2025 by bentherien
3
the gradient of bias in jax.nn.dot_product_attention bug Something isn't working
#27184 opened Mar 16, 2025 by bojone
1
GPU Kernel Cache Error When Using jax.random.permutation with Size=2 bug Something isn't working
#27183 opened Mar 16, 2025 by alik-git
2
Add static_argnames support in jax.checkpoint, as in jax.jit enhancement New feature or request
#27153 opened Mar 14, 2025 by oliverdutton-iso
1
jax-metal: jaxlib.xla_extension.XlaRuntimeError Apple GPU (Metal) plugin bug Something isn't working
#27146 opened Mar 14, 2025 by xuhengdada
@kulinseth @shuhand0
ProTip! Adding no:label will show everything without a label.