Skip to content

Issues: google/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

Clear current search query, filters, and sorts
35 Open 48 Closed
35 Open 48 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.experimental.enable_x64 fails in JIT context bug Something isn't working
#5982 opened Mar 8, 2021 by jakevdp updated Apr 8, 2021
@jakevdp
4
Support for AMD Radeon GPU on Mac OS Apple GPU (Metal) plugin contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. 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
#7163 opened Jul 1, 2021 by dbl001 updated Aug 22, 2023
6
PyInstaller package jax error bug Something isn't working contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#17705 opened Sep 21, 2023 by ysyecust updated Oct 22, 2023
8
segfault when using ensure_compile_time_eval bug Something isn't working
#18831 opened Dec 5, 2023 by benjaminvatterj updated Dec 5, 2023
1
Limit number of threads when running on CPU documentation
#1539 opened Oct 21, 2019 by dionhaefner updated Feb 7, 2024
29
Metal: reduction "Runtime canonicalization must simplify reduction axes to minor 4 dimensions" Apple GPU (Metal) plugin bug Something isn't working
#20112 opened Mar 7, 2024 by dlwh updated Mar 12, 2024
@kulinseth @shuhand0
4
Buffer donation for Metal plugin Apple GPU (Metal) plugin enhancement New feature or request
#20212 opened Mar 12, 2024 by dlwh updated Mar 13, 2024
@kulinseth @shuhand0
1
jax-metal: Failed assertion...expected element type f32 but received si32 Apple GPU (Metal) plugin bug Something isn't working
#19841 opened Feb 16, 2024 by csparker247 updated Mar 13, 2024
@kulinseth @shuhand0
2
Metal: missing functionality Apple GPU (Metal) plugin enhancement New feature or request
#20375 opened Mar 21, 2024 by Gattocrucco updated Mar 25, 2024
@shuhand0
1
Metal: failed to legalize operation 'mhlo.dot_general' for einsum "ijk,kji->k" Apple GPU (Metal) plugin bug Something isn't working
#20114 opened Mar 7, 2024 by dlwh updated Mar 26, 2024
@kulinseth @shuhand0
3
Metal: Failed Assertion / Crash when trying to create arrays in double-precision Apple GPU (Metal) plugin bug Something isn't working
#16435 opened Jun 15, 2023 by johnjmolina updated Mar 26, 2024
@kulinseth @shuhand0
8
Apple Metal: tile aborting kernel with error: 'anec.reshape' op result #0 must be 4D/5D memref... Apple GPU (Metal) plugin bug Something isn't working
#20413 opened Mar 24, 2024 by muffin-rice updated Mar 27, 2024
@shuhand0
1
Jax metal failed to install Apple GPU (Metal) plugin bug Something isn't working
#19886 opened Feb 20, 2024 by Kubiczek36 updated Apr 7, 2024
@kulinseth @shuhand0
8
On jax-metal, updating multidimensional boolean arrays sometimes fails Apple GPU (Metal) plugin bug Something isn't working
#20675 opened Apr 10, 2024 by shawwn updated Apr 10, 2024
@shuhand0
2
jax-metal: XlaRuntimeError: INTERNAL: Unable to serialize MPS module Apple GPU (Metal) plugin bug Something isn't working
#20401 opened Mar 22, 2024 by jsakaya updated Apr 22, 2024
@kulinseth @shuhand0
5
Apple Silicon: error: failed to legalize operation 'mhlo.triangular_solve' Apple GPU (Metal) plugin bug Something isn't working
#17490 opened Sep 7, 2023 by yu-fz updated Apr 24, 2024
@kulinseth @shuhand0
7
jax.numpy.insert returning incorrect results wen jitted on Metal Apple GPU (Metal) plugin bug Something isn't working
#20918 opened Apr 24, 2024 by fkeruzore updated Apr 25, 2024
@shuhand0
1
jax-metal: Memory leak on jit boundary, MPS Apple GPU (Metal) plugin bug Something isn't working
#20296 opened Mar 18, 2024 by youurayy updated May 19, 2024
@kulinseth @shuhand0
5
jax-metal: reduce window not supported Apple GPU (Metal) plugin bug Something isn't working
#21387 opened May 23, 2024 by jonatanklosko updated May 23, 2024
@kulinseth @shuhand0
jax-metal: count leading zeros not supported Apple GPU (Metal) plugin bug Something isn't working
#21389 opened May 23, 2024 by jonatanklosko updated May 23, 2024
@kulinseth @shuhand0
2
jax-metal: inconsistent behaviour of overflowing update slice in jax.lax.dynamic_update_slice Apple GPU (Metal) plugin bug Something isn't working
#21392 opened May 23, 2024 by jonatanklosko updated May 23, 2024
@shuhand0
jax-metal: inconsistent index clipping in jax.lax.dynamic_slice Apple GPU (Metal) plugin bug Something isn't working
#21393 opened May 23, 2024 by jonatanklosko updated May 23, 2024
@shuhand0
jax-metal: top-k invalid behaviour with NaNs Apple GPU (Metal) plugin bug Something isn't working
#21397 opened May 23, 2024 by jonatanklosko updated May 23, 2024
@kulinseth @shuhand0
jax-metal: reduce with multiple operands failed to legalize Apple GPU (Metal) plugin bug Something isn't working
#21384 opened May 23, 2024 by jonatanklosko updated May 29, 2024
@kulinseth @shuhand0
2
jax-metal: non-deterministic behavior of jnp.take_along_axis Apple GPU (Metal) plugin bug Something isn't working
#21757 opened Jun 10, 2024 by jkarwowski updated Jun 11, 2024
@kulinseth @shuhand0
2
ProTip! What’s not been updated in a month: updated:<2024-08-13.