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,482 Open 4,371 Closed
1,482 Open 4,371 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 compilation is slower with MPI bug Something isn't working
#25848 opened Jan 12, 2025 by SUSYUSTC
Tracer escaping in linalg.solve with ensure_compile_time_eval as of jax 0.4.36 bug Something isn't working
#25847 opened Jan 11, 2025 by mfschubert
'+ptx85' is not a recognized feature for this target (ignoring feature) ==> this issue occurs again bug Something isn't working
#25846 opened Jan 11, 2025 by jacquesmassa
jacrev, jacfwd and grad do not propagate sharding bug Something isn't working
#25844 opened Jan 10, 2025 by ASKabalan
@yashk2810
1
Better jax.debug.print enhancement New feature or request
#25842 opened Jan 10, 2025 by yliu120
@dfm
2
Got an error when offloading SplashAttention pallas call bug Something isn't working
#25841 opened Jan 10, 2025 by hanzhi713
RnnTest.test_struct_encoding_determinism failing bug Something isn't working
#25825 opened Jan 10, 2025 by belitskiy
2
Unable to Run JAX on GPU, can't find GPU device question Questions for the JAX team
#25822 opened Jan 10, 2025 by drewjin
3
Performance degradation of fftn with the new CPU runtime bug Something isn't working
#25808 opened Jan 9, 2025 by roth-jakob
@dfm
2
build error error: unknown type name 'svuint8_t' bug Something isn't working
#25799 opened Jan 9, 2025 by JaneConan
Cache initialization fails when a JAX Array is created before enabling local cache bug Something isn't working
#25768 opened Jan 8, 2025 by flferretti
@skye
1
Registering generic pytree nodes via getattr() and __setattr__ enhancement New feature or request
#25760 opened Jan 7, 2025 by shoyer
2
XLA runtime error when taking grad + vmap + scan on GPU bug Something isn't working
#25759 opened Jan 7, 2025 by EddieCunningham
1
vmap(jnp.asarray)(numpy_array) does not return a JAX array bug Something isn't working
#25745 opened Jan 7, 2025 by shoyer
1
@jakevdp
1
Bfloat16 jnp.allclose raises Illegal Instruction Exception bug Something isn't working
#25730 opened Jan 6, 2025 by aturker-synnada
3
vmap(custom_jvp) does not strip zeros from nondifferentiable return values, leading to AD crashes bug Something isn't working
#25724 opened Jan 5, 2025 by patrick-kidger
@dfm
1
If CUDA 12.1 is installed, pip-installed ptxas binary is not used and jax throws an error bug Something isn't working
#25718 opened Jan 3, 2025 by takkyu2
3
GPU pallas_call loses compiler params during second call when double jit-wrapped bug Something isn't working
#25714 opened Jan 3, 2025 by hanzhi713
@justinjfu
1
Quickstart: using JIT-compiled functions, jax.vmap does not provide significant speedup over naive implementation question Questions for the JAX team
#25703 opened Dec 31, 2024 by zsamboki
@jakevdp
1
Get "invalid value (nan) encountered in jit" even when jit disabled globally bug Something isn't working
#25701 opened Dec 31, 2024 by SUSYUSTC
@emilyfertig
7
jax.experimental.jet.jet series argument: unintuitive ordering; at least documentation should explain it better bug Something isn't working
#25700 opened Dec 31, 2024 by tjltjl
@mattjj
Batched meshgrid or alternative enhancement New feature or request
#25696 opened Dec 30, 2024 by Qazalbash
1
debugging/make_color_iter() returns the same index when the number of colors is 2 and do not use all the possible colors bug Something isn't working
#25695 opened Dec 30, 2024 by jeromemassot
Unexpected NaN gradient of jnp.abs at ±inf + 0j bug Something isn't working
#25681 opened Dec 24, 2024 by lfaucheux
@dfm
3
Unexpected NaN when signing ±inf + 0j bug Something isn't working
#25679 opened Dec 23, 2024 by lfaucheux
@jakevdp
4
ProTip! Add no:assignee to see everything that’s not assigned.