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,475 Open 4,392 Closed
1,475 Open 4,392 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

lambda x: jnp.log(x == 1) is not jit-able on metal. Apple GPU (Metal) plugin bug Something isn't working
#25935 opened Jan 16, 2025 by tillahoffmann
@kulinseth @shuhand0
Tracing performance: avoid jnp.asarray in library functions performance make things lean and fast
#25931 opened Jan 16, 2025 by jakevdp
@jakevdp
Different results in bf16/f32 mixed precision in conv_general_dilated in jax2tf bug Something isn't working
#25873 opened Jan 14, 2025 by bondquant
@gnecula
Add axis argument to lax.top_k enhancement New feature or request
#25867 opened Jan 14, 2025 by carlosgmartin
JAX getting stuck with asynchronous dispatching (CPU backend) bug Something isn't working
#25861 opened Jan 13, 2025 by roth-jakob
@dfm
12
ptxas unsupported version error bug Something isn't working
#25853 opened Jan 13, 2025 by lengstrom
@dfm
9
jax compilation is slower with MPI bug Something isn't working
#25848 opened Jan 12, 2025 by SUSYUSTC
1
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
@dougalm
1
'+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
3
Better jax.debug.print enhancement New feature or request
#25842 opened Jan 10, 2025 by yliu120
@dfm
3
Got an error when offloading SplashAttention pallas call bug Something isn't working
#25841 opened Jan 10, 2025 by hanzhi713
@justinjfu
RnnTest.test_struct_encoding_determinism failing bug Something isn't working
#25825 opened Jan 10, 2025 by belitskiy
2
Performance degradation of fftn with the new CPU runtime bug Something isn't working
#25808 opened Jan 9, 2025 by roth-jakob
@dfm
3
build error error: unknown type name 'svuint8_t' bug Something isn't working
#25799 opened Jan 9, 2025 by JaneConan
@jakevdp
3
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
2
Registering generic pytree nodes via getattr() and __setattr__ enhancement New feature or request
#25760 opened Jan 7, 2025 by shoyer
7
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
@dfm
5
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.