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,483 Open 4,406 Closed
1,483 Open 4,406 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

tests/pjit_test.py::PJitErrorTest::testAxisResourcesMismatch fails with "SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated!" bug Something isn't working
#26106 opened Jan 26, 2025 by booxter
Support for macOS 86_64 in README.md should be changed to something other than yes bug Something isn't working
#26105 opened Jan 26, 2025 by soraros
Exceptions from jax.pure_callback have no __cause__ bug Something isn't working
#26102 opened Jan 25, 2025 by crusaderky
printing new style prng keys on the cpu results in XlaRuntimeError bug Something isn't working
#26101 opened Jan 25, 2025 by SimonKoop
@jakevdp
8
Suppress XLA slow operation alarm enhancement New feature or request
#26098 opened Jan 25, 2025 by cool-RR
except A | B fails with jax.errors inside the JIT bug Something isn't working
#26088 opened Jan 24, 2025 by crusaderky
@jakevdp
7
Custom primitives + ordered effects + linear_solve = bug in 0.5 ? bug Something isn't working
#26087 opened Jan 24, 2025 by PhilipVinc
@sharadmv
1
Jax 0.5.0 error: no type named 'string' in namespace 'std' bug Something isn't working
#26062 opened Jan 23, 2025 by gongomgra
@penpornk
1
RET_CHECK failure using odeint with simple case bug Something isn't working
#26051 opened Jan 22, 2025 by DanPuzzuoli
@jakevdp
1
zeros_like does not match device when used under JIT bug Something isn't working
#26040 opened Jan 22, 2025 by jakevdp
@yashk2810
3
'arith.index_cast' op operand #0 must be signless-integer-like or memref of signless-integer, but got 'f32' bug Something isn't working
#26034 opened Jan 22, 2025 by Akshat-Tripathi
@sharadmv @justinjfu
2
CPU Slowdown introduced in 0.4.32 and in following versions bug Something isn't working
#26021 opened Jan 21, 2025 by danielpmorton
2
Pallax dot_general lowering bug for vectors bug Something isn't working
#26013 opened Jan 21, 2025 by LasseBlaauwbroek
@justinjfu
1
Provide API to query if jax.distributed has been initialised enhancement New feature or request
#26009 opened Jan 21, 2025 by PhilipVinc
@skye
Missing .device attribute inside @jax.jit bug Something isn't working
#26000 opened Jan 21, 2025 by crusaderky
18
Relax the transfer guard enhancement New feature or request type:performance
#25998 opened Jan 21, 2025 by crusaderky
@hawkinsp
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 NVIDIA GPU Issues specific to NVIDIA GPUs
#25986 opened Jan 20, 2025 by liamclarkza
4
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
1
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
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
ProTip! Add no:assignee to see everything that’s not assigned.