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

Clear current search query, filters, and sorts
41 Open 49 Closed
41 Open 49 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

Limit number of threads when running on CPU documentation
#1539 opened Oct 21, 2019 by dionhaefner
29
Apple Silicon: error: failed to legalize operation 'mhlo.cholesky' Apple GPU (Metal) plugin enhancement New feature or request
#16321 opened Jun 8, 2023 by adam-hartshorne
@kulinseth @shuhand0
18
Metal: add support for complex-valued math Apple GPU (Metal) plugin enhancement New feature or request
#16416 opened Jun 15, 2023 by hawkinsp
@kulinseth @shuhand0
17
Unable to initialize backend 'METAL' Apple GPU (Metal) plugin bug Something isn't working
#21383 opened May 23, 2024 by drbenvincent
@kulinseth @shuhand0
16
jax.lax.scan() segmentation fault with jax-metal on Mac M1 Apple GPU (Metal) plugin bug Something isn't working
#20750 opened Apr 14, 2024 by kdesoto-astro
@shuhand0
12
Apple Silicon: error: failed to legalize operation 'mhlo.custom_call' Apple GPU (Metal) plugin enhancement New feature or request
#16287 opened Jun 7, 2023 by mehdiataei
@kulinseth @shuhand0
11
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
@kulinseth @shuhand0
9
Jax metal failed to install Apple GPU (Metal) plugin bug Something isn't working
#19886 opened Feb 20, 2024 by Kubiczek36
@kulinseth @shuhand0
8
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
@kulinseth @shuhand0
8
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
8
jax-metal: Memory leak on jit boundary, MPS Apple GPU (Metal) plugin bug Something isn't working
#20296 opened Mar 18, 2024 by youurayy
@kulinseth @shuhand0
7
jax-metal: segmentation fault inside jax.lax.while_loop Apple GPU (Metal) plugin bug Something isn't working
#21552 opened May 31, 2024 by jonatanklosko
@shuhand0
6
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
6
JRT_Layouts_Extension is not found when PjRtCApiBuffer::layout is called when running basic example Apple GPU (Metal) plugin bug Something isn't working
#23820 opened Sep 21, 2024 by joaogui1
@shuhand0
5
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
@kulinseth @shuhand0
5
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
@kulinseth @shuhand0
4
jax.experimental.enable_x64 fails in JIT context bug Something isn't working
#5982 opened Mar 8, 2021 by jakevdp
@jakevdp
4
error: failed to legalize operation 'mhlo.erf' with jax-metal bug Something isn't working
#23384 opened Sep 2, 2024 by a-gn
@kulinseth @shuhand0
3
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
@kulinseth @shuhand0
3
FlaxLlamaForCausalLMModule hanging on jax-metal bug Something isn't working
#24221 opened Oct 9, 2024 by alexlatif
2
Metal: missing functionality Apple GPU (Metal) plugin enhancement New feature or request
#20375 opened Mar 21, 2024 by Gattocrucco
@shuhand0
2
jax-metal: count leading zeros not supported Apple GPU (Metal) plugin bug Something isn't working
#21389 opened May 23, 2024 by jonatanklosko
@kulinseth @shuhand0
2
jax-metal: cond fails in compile in certain cases Apple GPU (Metal) plugin bug Something isn't working
#21601 opened Jun 3, 2024 by jonatanklosko
@shuhand0
2
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
@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
@kulinseth @shuhand0
2
ProTip! no:milestone will show everything without a milestone.