Skip to content

Issues: jax-ml/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,415 Open 4,247 Closed
1,415 Open 4,247 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

JAX RNG clashes with PyTorch on GPU bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#15084 opened Mar 19, 2023 by murphyk
6 😕 1
Improved error messages and documentation regarding what can/cannot be jited documentation
#953 opened Jul 1, 2019 by samuela
@mattjj
7
jax.numpy array indexing has different out-of-bounds behavior to numpy documentation NVIDIA GPU Issues specific to NVIDIA GPUs P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#278 opened Jan 26, 2019 by jonasrauber
@hawkinsp
16
jit cache leaks memory bug Something isn't working
#282 opened Jan 27, 2019 by jonasrauber
@hawkinsp
4
document when and how to make ndarray copies of DeviceArrays documentation
#390 opened Feb 16, 2019 by mattjj
2
PyPy support enhancement New feature or request
#474 opened Mar 3, 2019 by hawkinsp
13
Implement JVP for SVD when full_matrices=True enhancement New feature or request
#508 opened Mar 14, 2019 by hawkinsp
9
Support autodiff of Eigendecomposition with repeated eigenvalues enhancement New feature or request
#669 opened May 4, 2019 by sethaxen
1
34
Jit with unknown shape question Questions for the JAX team
#803 opened Jun 2, 2019 by guyvdbroeck
@mattjj
4
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
document memory management (DeviceArrays and where to find them) documentation enhancement New feature or request
#863 opened Jun 17, 2019 by mattjj
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
Linearity (or affineness?) checking for PRNG keys enhancement New feature or request
#192 opened Jan 3, 2019 by j-towns
@jakevdp
8
slow pmap allreduce enhancement New feature or request P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional) performance make things lean and fast
#973 opened Jul 4, 2019 by christopherhesse
@hawkinsp
7
feature request: sparse jacobian and sparse hessians enhancement New feature or request
#1032 opened Jul 18, 2019 by martinResearch
23
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
Functions like jax.numpy.concatenate are not compatible with jax.sharding, resulting in improper functioning bug Something isn't working
#24576 opened Oct 29, 2024 by kYangLi
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
Provide filename and line number for the compiling function in compilation logging enhancement New feature or request
#1247 opened Aug 26, 2019 by zhongwen
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
@dfm
31
Custom tangent types for gradient computations enhancement New feature or request
#918 opened Jun 25, 2019 by jheek
2
ProTip! Find all open issues with in progress development work with linked:pr.