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,353 Open 3,930 Closed
1,353 Open 3,930 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

Is there difference in computing gradient between jax and torch bug Something isn't working
#22101 opened Jun 26, 2024 by 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
Array dispatching with __array_ufunc__ in JAX enhancement New feature or request
#22090 opened Jun 25, 2024 by gautierronan
@jakevdp
7
jaxlib or libtpu not detected on TPU Pod bug Something isn't working
#22070 opened Jun 24, 2024 by ayaka14732
@skye
7
Command to build jaxlib? question Questions for the JAX team
#22046 opened Jun 22, 2024 by yurivict
@jakevdp
1
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
shard_map should support static_argnums enhancement New feature or request
#22043 opened Jun 22, 2024 by reinerp
1
@mattjj
4
jaxlib wheel hardcoded to manylinux2014 platform bug Something isn't working
#22034 opened Jun 21, 2024 by mrodden
@hawkinsp
3
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
reshape operation with Pallas on TPU/Mosaic cause XlaRuntimeError: INTERNAL bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#21999 opened Jun 20, 2024 by Lime-Cakes
1
jax.xla_computation as_hlo_text() enhancement New feature or request
#21990 opened Jun 20, 2024 by aoshujiaocheng
2
T5x+jax pretraining longt5xl-3B 4096 input oom bug Something isn't working
#21985 opened Jun 20, 2024 by robotzheng
3
RuntimeError: jaxlib/gpu/solver_kernels.cc:45: operation gpusolverDnCreate(&handle) failed: cuSolver internal error bug Something isn't working
#21950 opened Jun 18, 2024 by seishiroono
@dfm
6
Import error bug Something isn't working needs info More information is required to diagnose & prioritize the issue.
#21940 opened Jun 18, 2024 by jntdst
1
sparse: bcoo_spdot_general have incorrect abstract_eval bug Something isn't working
#21921 opened Jun 17, 2024 by Alan-Chen99
1
@jakevdp
1
Docs for persistent compilation cache are superseded bug Something isn't working documentation
#21908 opened Jun 17, 2024 by epignatelli
@ayaka14732
10
Is there a way to control the pjit to dynamically partition the input IDs into different ranks? needs info More information is required to diagnose & prioritize the issue. question Questions for the JAX team
#21902 opened Jun 16, 2024 by MoFHeka
2 tasks done
8
Edge behavior in jax.scipy.special.betainc bug Something isn't working
#21900 opened Jun 15, 2024 by mdhaber
@jakevdp
6
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
jnp fails after a = np.zeros((100, 100), dtype=np.int4) on mac m1 bug Something isn't working
#21883 opened Jun 14, 2024 by Maxon081102
1
Unexpected AllReduce in backward pass with shard_map, custom_vjp, and pallas bug Something isn't working
#21855 opened Jun 13, 2024 by southfreebird
@mattjj
1
jnp fails after enable x64 with Apple M1 chip Apple GPU (Metal) plugin bug Something isn't working duplicate This issue or pull request already exists
#21840 opened Jun 12, 2024 by willcai7
2
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
ProTip! Updated in the last three days: updated:>2024-06-23.