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,549 Open 4,548 Closed
1,549 Open 4,548 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

Jitting a function with enable_x64 does not work for jnp.linalg.solve and related functions. bug Something isn't working
#27602 opened Mar 30, 2025 by mishavanbeek
jax.lax.scan does not allow static loop variables for computing static values (e.g. for jnp.pad) enhancement New feature or request
#27601 opened Mar 30, 2025 by vietluong2110
jax.jit has general issues with gfx906 [ROCM] bug Something isn't working
#27600 opened Mar 30, 2025 by zmelumian972
cudnn dot_product_attention encounters Failed to capture gpu graph when running on 2 NVIDIA A6000 Ada cards bug Something isn't working
#27599 opened Mar 30, 2025 by Methylamphetamine
Use MiniDebugInfo debug information on Linux wheel shared libraries to reduce wheel size enhancement New feature or request type:others
#27598 opened Mar 30, 2025 by hawkinsp
Clarify whether lax.top_k returns results in sorted order enhancement New feature or request
#27594 opened Mar 30, 2025 by carlosgmartin
jax.vmap and jax.lax.map result changes drastically with large array/batch_size bug Something isn't working
#27591 opened Mar 29, 2025 by gbuzzard
1
1
Remove nanobind pin in FFI example requirements cleanup Code cleanups and improvements
#27563 opened Mar 28, 2025 by danielsuo
@danielsuo
jax.jit(..., inline=True) converts arrays back into Python scalars. bug Something isn't working
#27545 opened Mar 27, 2025 by patrick-kidger
1
Jax is unable to findn and uze the gpu bug Something isn't working
#27535 opened Mar 27, 2025 by Its-a-me-Ashwin
[mosaic] Indexing into 3rd minor-most+ dimensionwith sub 32-bit dtypes enhancement New feature or request
#27532 opened Mar 27, 2025 by oliverdutton-iso
Lower Precision Grad Reduction bug Something isn't working
#27496 opened Mar 26, 2025 by jfc4050
@mattjj
2
Implementation of scipy.integrate.quad into jax.scipy enhancement New feature or request
#27493 opened Mar 26, 2025 by szoppelt
2 tasks done
@jakevdp
1
[Pallas] support negative indexing of array refs enhancement New feature or request
#27490 opened Mar 26, 2025 by ysngshn
4
error_check API does not support usage within vmap bug Something isn't working
#27483 opened Mar 26, 2025 by emergenz
@mattjj @dougalm @ayaka14732
Global mutable arrays not supported in custom_jvp bug Something isn't working
#27470 opened Mar 26, 2025 by ayaka14732
@dougalm
Schrödinger's JAX (using jax.debug.print fixes wrong results introduced in 0.5.3) bug Something isn't working
#27464 opened Mar 26, 2025 by gboehl
15
Array construction on CPU is 200x slower than NumPy bug Something isn't working
#27460 opened Mar 26, 2025 by sbodenstein
1
@superbobry
3
[Splash Attention] Remove unnecessary head_dim_v % NUM_LANES == 0 constraint enhancement New feature or request
#27427 opened Mar 25, 2025 by ds-hwang
Relax assumptions for bcoo_dot_general enhancement New feature or request P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#27397 opened Mar 25, 2025 by ryan112358
2
Vmap quickstart tutorial produces an assert error bug Something isn't working
#27362 opened Mar 24, 2025 by mickeybeurskens
@dfm
4
scipy support via Array API meta-issue enhancement New feature or request
#27361 opened Mar 24, 2025 by crusaderky
3
Timeouts in TSAN CI job under Free-threading bug Something isn't working
#27360 opened Mar 24, 2025 by vfdev-5
Out of GPU memory bug Something isn't working
#27354 opened Mar 23, 2025 by yang-yangfeng
@dfm
1
FFI Custom JVP + VJP Support enhancement New feature or request
#27352 opened Mar 22, 2025 by Edenhofer
3
ProTip! Find all open issues with in progress development work with linked:pr.