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

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: GPU crash(?) with large input Apple GPU (Metal) plugin bug Something isn't working
#23902 opened Sep 25, 2024 by matthewlai
@shuhand0
1
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
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
jax-metal slowdown on m3 max when external monitor is plugged in Apple GPU (Metal) plugin bug Something isn't working
#24163 opened Oct 7, 2024 by bjeurissen
@shuhand0
1
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-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
Segmentation Fault in jax-metal with variable indexing and partial slice in scan Apple GPU (Metal) plugin bug Something isn't working
#23931 opened Sep 25, 2024 by bsarkar321
jax-metal: reduce window not supported Apple GPU (Metal) plugin bug Something isn't working
#21387 opened May 23, 2024 by jonatanklosko
@kulinseth @shuhand0
Jax metal failed to install Apple GPU (Metal) plugin bug Something isn't working
#19886 opened Feb 20, 2024 by Kubiczek36
@kulinseth @shuhand0
8
Buffer donation for Metal plugin Apple GPU (Metal) plugin enhancement New feature or request
#20212 opened Mar 12, 2024 by dlwh
@kulinseth @shuhand0
1
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: 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
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: top-k invalid behaviour with NaNs Apple GPU (Metal) plugin bug Something isn't working
#21397 opened May 23, 2024 by jonatanklosko
@kulinseth @shuhand0
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
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: 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
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
@shuhand0
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
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
@shuhand0
2
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
@shuhand0
1
jax-metal: minimum/maximum/min/max/argmin/argmax invalid output with NaNs Apple GPU (Metal) plugin bug Something isn't working
#21821 opened Jun 12, 2024 by jonatanklosko
@kulinseth @shuhand0
ProTip! Mix and match filters to narrow down what you’re looking for.