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

[Apple GPU (Metal) plugin] (Stable) HLO support ? Apple GPU (Metal) plugin enhancement New feature or request
#24365 by janpfeifer was closed Oct 18, 2024
3
jax-metal on m1 couldn't compile random.PRNGKey properly Apple GPU (Metal) plugin bug Something isn't working
#21783 by ziweiyang was closed Jun 11, 2024
@kulinseth @shuhand0
1
jax-metal: dynamic update slice fails with unsigned indices Apple GPU (Metal) plugin bug Something isn't working
#21590 by jonatanklosko was closed Jun 12, 2024
@kulinseth @shuhand0
1
jax-metal: argmax fails with unsigned index type Apple GPU (Metal) plugin bug Something isn't working
#21577 by jonatanklosko was closed Jun 12, 2024
@shuhand0
1
jax-metal: 3-D convolution crashes Apple GPU (Metal) plugin bug Something isn't working
#21554 by jonatanklosko was closed Jun 12, 2024
@shuhand0
1
jax-metal: gather fails with unsigned indices Apple GPU (Metal) plugin bug Something isn't working
#21547 by jonatanklosko was closed Jun 12, 2024
@shuhand0
1
jax/jaxlib not working bug Something isn't working
#21491 by RivasLab was closed Jun 6, 2024
@dfm
6
Metal : fp64 operations with jax.numpy base functions not supported Apple GPU (Metal) plugin bug Something isn't working duplicate This issue or pull request already exists
#20938 by aboucaud was closed Jun 13, 2024
@kulinseth @shuhand0
4
Probability density/mass functions from jax.scipy.stats not supported on Metal Apple GPU (Metal) plugin bug Something isn't working
#20835 by logan-blaine was closed May 28, 2024
@kulinseth @shuhand0
2
jax.numpy.mean ops convert to tflite is slowly
#20672 by noahzhy was closed Apr 11, 2024
@mattjj
1
jax-metal: current mps dialect version is 1.0.0, can't parse version 1.1.0 Apple GPU (Metal) plugin bug Something isn't working
#20338 by tillahoffmann was closed Mar 21, 2024
@shuhand0
6
error: failed to legalize operation 'mhlo.pad' Apple GPU (Metal) plugin bug Something isn't working
#20119 by nullhook was closed Mar 12, 2024
@kulinseth @shuhand0
4
Metal: failed assertion New volume: 6 should match old volume: 24 for einsum "ijk,kji->kij" Apple GPU (Metal) plugin bug Something isn't working
#20115 by dlwh was closed Mar 12, 2024
@kulinseth @shuhand0
2
Metal: "mhlo.pad" error when calling jax.numpy.cumsum Apple GPU (Metal) plugin bug Something isn't working
#19402 by dkgaraujo was closed Mar 12, 2024
@kulinseth @shuhand0
2
jax-metal, M1: failed to materialize conversion for result #0 of operation 'mhlo.convolution' Apple GPU (Metal) plugin bug Something isn't working
#19369 by a-gn was closed May 5, 2024
@kulinseth @shuhand0
4
Potential memory issue on Apple Metal Apple GPU (Metal) plugin bug Something isn't working
#19132 by milutter was closed Apr 23, 2024
@kulinseth @shuhand0
3
[METAL] GPU Timeout error bug Something isn't working
#19110 by tw-ilson was closed Dec 23, 2023
1
Failed to legalize rounding operation in jax-metal Apple GPU (Metal) plugin bug Something isn't working
#18678 by ghcollin was closed Mar 12, 2024
@kulinseth @shuhand0
5
Trying to use Objax on Jax-metal Apple GPU (Metal) plugin bug Something isn't working
#18674 by silvialpz was closed May 28, 2024
@kulinseth @shuhand0
1
Issue with JAX Installation Verification on M2 Apple GPU (Metal) plugin bug Something isn't working
#18598 by UnicornHJ was closed Nov 19, 2023
@kulinseth @shuhand0
RuntimeError in jax-metal Apple GPU (Metal) plugin bug Something isn't working
#18006 by eterevsky was closed Oct 23, 2023
@kulinseth @shuhand0
4
Attribute error on import: module 'ml_dtypes' has no attribute 'float8_e4mb11' bug Something isn't working
#17693 by priyakasimbeg was closed May 8, 2024
@jakevdp
8
jaxlib compilation error on 0.4.11 bug Something isn't working
#17630 by stenczelt was closed Sep 22, 2023
2
Incorrect result for in1d on Apple GPU using jax-metal Apple GPU (Metal) plugin bug Something isn't working
#17590 by sfc-gh-arozenshteyn was closed Jan 2, 2024
@kulinseth @shuhand0
2
Metal backend leads to NaN with take_along_axis Apple GPU (Metal) plugin bug Something isn't working
#17344 by tdgfrost was closed Jan 2, 2024
@kulinseth @shuhand0
2
ProTip! Type g i on any issue or pull request to go back to the issue listing page.