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,476 Open 4,395 Closed
1,476 Open 4,395 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

Missing .device attribute inside @jax.jit bug Something isn't working
#26000 opened Jan 21, 2025 by crusaderky
1
Relax the transfer guard enhancement New feature or request type:performance
#25998 opened Jan 21, 2025 by crusaderky
jax.scipy.linalg.expm yields wrong values (for symmetric matrices) bug Something isn't working
#25987 opened Jan 20, 2025 by Ferlemann
jax.nn.dot_product_attention CuDNN implementation raises tensor stride error during jit compile bug Something isn't working
#25986 opened Jan 20, 2025 by liamclarkza
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
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.