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

1,339 Open 3,898 Closed
1,339 Open 3,898 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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-metal on m1 couldn't compile random.PRNGKey properly bug Something isn't working
#21783 opened Jun 10, 2024 by ziweiyang
jax.scipy.special.expi extremely slow when applied to a jnp.array bug Something isn't working
#21760 opened Jun 10, 2024 by AntLrc
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
1
TracerBoolConversionError when jitting jax.numpy.linalg.norm bug Something isn't working
#21730 opened Jun 7, 2024 by mvilucchio
@jakevdp
1
partial eval silently skips effects question Questions for the JAX team
#21713 opened Jun 6, 2024 by Alan-Chen99
@mattjj
2
ndarray.at.set(mode="drop") gives incorrect value for the last element in the array bug Something isn't working
#21708 opened Jun 6, 2024 by Gregory-Meyer
@jakevdp
4
pmap out_axes=None doesn't check if output is mapped/unmapped bug Something isn't working P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#21671 opened Jun 5, 2024 by mattjj
Inconsistent results with shard_map when switching PRNG Implementation from threefry2x32 to rbg documentation
#21663 opened Jun 5, 2024 by southfreebird
@froystig
1
Using jax.config.update within config context manager fails to set new value bug Something isn't working needs info More information is required to diagnose & prioritize the issue.
#21650 opened Jun 4, 2024 by jakevdp
@hawkinsp
7
Wrong array values in jax.debug.print and the actual results when using lax.slice in lax.scan bug Something isn't working
#21637 opened Jun 4, 2024 by mfkasim1
@superbobry
5
jaxlib.xla_extension.XlaRuntimeError: INTERNAL: ptxas exited with non-zero error code 11 bug Something isn't working
#21621 opened Jun 4, 2024 by PhilipVinc
2
support batched matrix multiplication in pallas enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#21618 opened Jun 4, 2024 by zhixuan-lin
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
1
jax-metal: dynamic update slice fails with unsigned indices Apple GPU (Metal) plugin bug Something isn't working
#21590 opened Jun 3, 2024 by jonatanklosko
@kulinseth @shuhand0
Guidelines on reducing compilation memory? enhancement New feature or request
#21583 opened Jun 2, 2024 by OhadRubin
str(PyTreeDef) identical for two PyTreeDefs, but assert with allclose fails bug Something isn't working
#21581 opened Jun 1, 2024 by johannahaffner
@superbobry
6
jax-metal: argmax fails with unsigned index type Apple GPU (Metal) plugin bug Something isn't working
#21577 opened Jun 1, 2024 by jonatanklosko
@shuhand0
Involuntary full rematerialization for advanced data movements enhancement New feature or request
#21562 opened May 31, 2024 by chaserileyroberts
4
constants in Pallas kernel causes ValueError: safe_zip() argument 2 is shorter than argument 1 bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#21557 opened May 31, 2024 by zhixuan-lin
@superbobry
2
jax-metal: 3-D convolution crashes Apple GPU (Metal) plugin bug Something isn't working
#21554 opened May 31, 2024 by jonatanklosko
@shuhand0
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
2
jax-metal: gather fails with unsigned indices Apple GPU (Metal) plugin bug Something isn't working
#21547 opened May 31, 2024 by jonatanklosko
@shuhand0
jax.scipy.special.hyp1f1 unstable where scipy.special.hyp1f1 is not bug Something isn't working
#21503 opened May 30, 2024 by HansN87
1
Cryptic error message when running pallas kernel on TPU bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#21488 opened May 29, 2024 by finbarrtimbers
4
Can´t get jax profiling to work bug Something isn't working
#21483 opened May 29, 2024 by MaHaWo
@hawkinsp
5
ProTip! What’s not been updated in a month: updated:<2024-05-10.