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

Clear current search query, filters, and sorts
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

Building on linux with ppc64le CPU contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome.
#4493 opened Oct 8, 2020 by f0uriest
74
BFGS/Quasi-Newton optimizers?
#1400 opened Sep 26, 2019 by proteneer
61
autodiff support for jax.numpy.linalg.eig enhancement New feature or request
#2748 opened Apr 17, 2020 by froystig
1
@j-towns
51
sparse eigenvalue solvers contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. enhancement New feature or request
#3112 opened May 15, 2020 by mganahl
1
47
Differentiable matrix-free linear algebra, optimization and equation solving enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#1531 opened Oct 20, 2019 by shoyer
6 tasks
38
TPU not working bug Something isn't working TPU Issues related to compilation for or performance on Google TPUs
#13260 opened Nov 15, 2022 by chrisflesher
@skye
37
Support autodiff of Eigendecomposition with repeated eigenvalues enhancement New feature or request
#669 opened May 4, 2019 by sethaxen
1
34
vmap of cond's predicate results in select, leading to unexpected compute/memory use enhancement New feature or request performance make things lean and fast
#8409 opened Oct 30, 2021 by aespielberg
@froystig @mattjj
32
any suggestions on how to improve performance for gradient step with odeint?
#3993 opened Aug 7, 2020 by ibulu
31
[README] nn libraries fix documentation
#20745 opened Apr 14, 2024 by keren9898
30
Extremely slow GPU execution bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs P1 (soon) Assignee is working on this now, among other tasks. (Assignee required)
#7024 opened Jun 18, 2021 by NeilGirdhar
@nouiz
30
jnp.linalg.svd etc. do not respect __jax_array__ contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. enhancement New feature or request
#10065 opened Mar 29, 2022 by patrick-kidger
@mattjj
29
More flexible ODE integration
#2628 opened Apr 7, 2020 by samuela
3 tasks
29
Limit number of threads when running on CPU documentation
#1539 opened Oct 21, 2019 by dionhaefner
29
Support (nonsymmetric) np.linalg.eig on GPU enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#1259 opened Aug 28, 2019 by clemisch
28
Can't build jaxlib in GH200 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21299 opened May 19, 2024 by giladqm
27
Automatically treat dataclasses as pytrees
#2371 opened Mar 6, 2020 by shoyer
1
25
Proposed Dockerfile for Running JAX enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#6340 opened Apr 4, 2021 by AndrewCSQ
@skye
23
Outstanding Primitive Rules for Higher Order Automatic Differentiation (jet)
#2431 opened Mar 16, 2020 by jessebett 100+
23
feature request: sparse jacobian and sparse hessians enhancement New feature or request
#1032 opened Jul 18, 2019 by martinResearch
23
[RFE] Add support for distributed CPU-backend mode CPU Issues related to the CPU compiler/runtime enhancement New feature or request
#11182 opened Jun 21, 2022 by PhilipVinc
@hawkinsp
22
Add mechanism to clear full compilation cache enhancement New feature or request
#10828 opened May 25, 2022 by kach
2
@mattjj
22
Clear GPU memory NVIDIA GPU Issues specific to NVIDIA GPUs P1 (soon) Assignee is working on this now, among other tasks. (Assignee required) question Questions for the JAX team
#1222 opened Aug 21, 2019 by clemisch
22
Need help understanding memory usage (CPU backend)
#5071 opened Dec 1, 2020 by swertz
21
[Proposal] Consistent argnums and argnames parameters for transformations enhancement New feature or request
#10614 opened May 6, 2022 by JeppeKlitgaard
15 tasks
20
ProTip! Adding no:label will show everything without a label.