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
736 Open 1,885 Closed
736 Open 1,885 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

Let's split the VJP for odeint into JVP and transpose
#1927 opened Dec 30, 2019 by duvenaud
1 of 8 tasks
6 🚀 6
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 🚀 5
Psuedoinverse solves enhancement New feature or request
#2802 opened Apr 23, 2020 by shoyer
8 🚀 4
[FR] add betaincinv enhancement New feature or request P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#2399 opened Mar 10, 2020 by tuchandra
15 🚀 2
Hot wiring xla'd kernels question Questions for the JAX team
#4866 opened Nov 10, 2020 by proteneer
4 🚀 2
Make softplus support beta parameter enhancement New feature or request
#18443 opened Nov 8, 2023 by e-pet
🚀 2
JEP: Allow finite difference method for non-jax differentiable functions enhancement New feature or request
#15425 opened Apr 6, 2023 by mbmccoy
12 🚀 2
Implement jax.scipy.linalg.logm contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. enhancement New feature or request open Issues intentionally left open, with no schedule for next steps.
#5469 opened Jan 20, 2021 by jaeyoo
18 🚀 1
Scanning over reduction rather than reducing over scan is a powerful optimisation - but hard to implement
#4968 opened Nov 19, 2020 by SamPruden
9 🚀 1
Improve compilation speed for generated codes with automatically marginalized MCMC enhancement New feature or request
#17335 opened Aug 28, 2023 by lll6924
2 tasks done
🚀 1
Implement JVP for SVD when full_matrices=True enhancement New feature or request
#508 opened Mar 14, 2019 by hawkinsp
9
Feature Request: Add a preduce_sum method enhancement New feature or request
#839 opened Jun 11, 2019 by chaserileyroberts
Feature request: allow passing auxiliary data when defining custom gradients
#844 opened Jun 12, 2019 by nkitaev
1
Allow to selectively execute ram-heavy ops on CPU 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)
#915 opened Jun 24, 2019 by romanngg
2
Custom tangent types for gradient computations enhancement New feature or request
#918 opened Jun 25, 2019 by jheek
2
The cost of index_update
#1034 opened Jul 19, 2019 by samuela
3
is keyword is not preserved under vmap. documentation
#1103 opened Aug 3, 2019 by sschoenholz
3
[question] Call overhead / disable parallelism
#1169 opened Aug 12, 2019 by aseyboldt
4
Feature request: extracting intermediate values from custom vjp enhancement New feature or request
#1197 opened Aug 16, 2019 by gehring
1
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
jax.numpy functions out argument fails question Questions for the JAX team
#1328 opened Sep 9, 2019 by mgbukov
3
Feature request: grad tracing could consider only the "critical" path
#1338 opened Sep 12, 2019 by schmrlng
3
cuDNN attention segmentation fault bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#23701 opened Sep 17, 2024 by sbodenstein
1
BFGS/Quasi-Newton optimizers?
#1400 opened Sep 26, 2019 by proteneer
61
[feature request] Provide access to Pytree to_iterable and from_iterable functions
#1527 opened Oct 18, 2019 by botev
2
ProTip! Adding no:label will show everything without a label.