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,334 Open 4,175 Closed
1,334 Open 4,175 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

Automatically differentiate Pallas kernel failed bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#22197 opened Jun 29, 2024 by Lime-Cakes
@superbobry
4
Efficient parallelization of tasks with highly variable difficulty and run times enhancement New feature or request
#22176 opened Jun 28, 2024 by javier-garcia-tilburg
2 tasks done
Error Building Jaxlib v0.4.30 on Jetson Orin bug Something isn't working
#22155 opened Jun 27, 2024 by jwknaup
17
Global Singleton (XlaDebugInfoManager) leaks out of the control of C API and gets two copies in two shared libraries bug Something isn't working
#22148 opened Jun 27, 2024 by yliu120
4
cuSolver internal error with jax.scipy.stats.multivariate_normal.pdf() bug Something isn't working
#22127 opened Jun 26, 2024 by jwknaup
int4 reshape: Reshape should have supported layout before reaching the emitter bug Something isn't working
#22121 opened Jun 26, 2024 by py4
Is there difference in computing gradient between jax and torch needs info More information is required to diagnose & prioritize the issue.
#22101 opened Jun 26, 2024 by ergo-zyh
@ergo-zyh
1
Figure out + document how to use manual capture profiling with Kubernetes documentation enhancement New feature or request
#22096 opened Jun 25, 2024 by skye
1
Array dispatching with __array_ufunc__ in JAX enhancement New feature or request
#22090 opened Jun 25, 2024 by gautierronan
@jakevdp
8
jaxlib or libtpu not detected on TPU Pod bug Something isn't working
#22070 opened Jun 24, 2024 by ayaka14732
@skye
9
scan of device_put carry raises TypeError during the backward pass bug Something isn't working
#22045 opened Jun 22, 2024 by fehiepsi
@yashk2810
2
lax.cond sometimes inserts a nonlinear lax.stop_gradient into its JVP rule. bug Something isn't working
#22011 opened Jun 20, 2024 by patrick-kidger
@mattjj
1
T5x+jax pretraining longt5xl-3B 4096 input oom bug Something isn't working
#21985 opened Jun 20, 2024 by robotzheng
3
Edge behavior in jax.scipy.special.betainc bug Something isn't working
#21900 opened Jun 15, 2024 by mdhaber
@jakevdp
7
forward and reverse mode gradients for eigvalsh + log or eigvalsh + sqrt returns NaNs bug Something isn't working
#21896 opened Jun 15, 2024 by h-vijayakumaran
@hawkinsp
1
Unexpected AllReduce in backward pass with shard_map, custom_vjp, and pallas enhancement New feature or request
#21855 opened Jun 13, 2024 by southfreebird
1
@mattjj
4
jit-eval_shape-<callable PyTree with numpy __eq__ semantics> crashes on JAX 0.4.29 bug Something isn't working
#21825 opened Jun 12, 2024 by patrick-kidger
@yashk2810
Tracers are now hashable. bug Something isn't working
#21824 opened Jun 12, 2024 by patrick-kidger
1
@jakevdp
12
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
jax.scipy.special.expi extremely slow when applied to a jnp.array bug Something isn't working
#21760 opened Jun 10, 2024 by AntLrc
3
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
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
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
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
ProTip! Add no:assignee to see everything that’s not assigned.