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,356 Open 3,937 Closed
1,356 Open 3,937 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
#22197 opened Jun 29, 2024 by Lime-Cakes
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
1
How to perform a Bucketize and dynamic_partition OP？ question Questions for the JAX team
#22151 opened Jun 27, 2024 by MoFHeka
2 tasks done
@jakevdp
5
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
1
Mark jax typing Protocols as @runtime_checkable enhancement New feature or request
#22144 opened Jun 27, 2024 by bionicles
2 tasks done
@superbobry
1
Wrong result for unsigned dtype input into jax.numpy.partition and jax.numpy.argpartition bug Something isn't working
#22137 opened Jun 27, 2024 by JuliaPoo
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 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
8
jaxlib or libtpu not detected on TPU Pod bug Something isn't working
#22070 opened Jun 24, 2024 by ayaka14732
@skye
8
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
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
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
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
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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.