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

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
extensions module enhancement New feature or request JEP JAX enhancement proposal
#15856 opened May 3, 2023 by froystig
@froystig @jakevdp
@sharadmv @yashk2810
3 🚀 4
Need xla_python_gpu_callback with dlpack interface enhancement New feature or request
#11371 opened Jul 5, 2022 by YouJiacheng
@sharadmv
6 🚀 4
Psuedoinverse solves enhancement New feature or request
#2802 opened Apr 23, 2020 by shoyer
8 🚀 4
[ROCm] JAX-ROCm docker images AMD GPU Issues pertaining to AMD GPUs (ROCM) enhancement New feature or request
#7598 opened Aug 12, 2021 by reza-amd
@yashk2810
15 🚀 3
JEP: Allow finite difference method for non-jax differentiable functions enhancement New feature or request
#15425 opened Apr 6, 2023 by mbmccoy
12 🚀 2
Hot wiring xla'd kernels question Questions for the JAX team
#4866 opened Nov 10, 2020 by proteneer
4 🚀 2
implement a wider PRNG enhancement New feature or request P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#7676 opened Aug 20, 2021 by froystig
@froystig
🚀 2
[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
Make softplus support beta parameter enhancement New feature or request
#18443 opened Nov 8, 2023 by e-pet
🚀 2
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
JEP: Unified Error Handling API JEP JAX enhancement proposal
#12537 opened Sep 27, 2022 by sharadmv
@sharadmv @mattjj @LenaMartens
1 🚀 1
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
jax.grad(jnp.log) returns non-nan for out-of-domain inputs bug Something isn't working
#13941 opened Jan 10, 2023 by pierreglaser
@jakevdp
1 🚀 1
Improve error messages for invalid indices better_errors Improve the error reporting
#22798 opened Jul 31, 2024 by jakevdp
@mattjj
🚀 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
Orthogonal Initializer raises gpusolverDnCreate(&handle) failed: cuSolver internal error bug Something isn't working
#23616 opened Sep 13, 2024 by brianorbrain
Performance Issues with Summing Per-Example Gradients (Caused by Fusions?) bug Something isn't working performance make things lean and fast
#1232 opened Aug 22, 2019 by lumip
1
Custom tangent types for gradient computations enhancement New feature or request
#918 opened Jun 25, 2019 by jheek
2
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
document memory management (DeviceArrays and where to find them) documentation enhancement New feature or request
#863 opened Jun 17, 2019 by mattjj
2
Feature request: allow passing auxiliary data when defining custom gradients
#844 opened Jun 12, 2019 by nkitaev
1
PyPy support enhancement New feature or request
#474 opened Mar 3, 2019 by hawkinsp
13
document when and how to make ndarray copies of DeviceArrays documentation
#390 opened Feb 16, 2019 by mattjj
2
ProTip! Find all open issues with in progress development work with linked:pr.