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,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

[feature request] Provide access to Pytree to_iterable and from_iterable functions
#1527 opened Oct 18, 2019 by botev
2
jax.ops.index_update memory usage blows up during derivative evaluation
#1484 opened Oct 10, 2019 by adabbott
@jekbradbury
1
Jax for Web? (JS api or web assembly guide) P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#1472 opened Oct 9, 2019 by bionicles
@gnecula
14
Allowing pmap on CPUs by making CPU cores visibles directly on Jax. CPU Issues related to the CPU compiler/runtime P1 (soon) Assignee is working on this now, among other tasks. (Assignee required)
#1408 opened Sep 27, 2019 by sguada
@skye
17
BFGS/Quasi-Newton optimizers?
#1400 opened Sep 26, 2019 by proteneer
61
jax.numpy.log(2., dtype=np.float32) does not accept dtype arg
#1364 opened Sep 17, 2019 by brianwa84
1
Remove the multiple_results field for core.Primitive cleanup Code cleanups and improvements
#1342 opened Sep 13, 2019 by gnecula
1
Feature request: grad tracing could consider only the "critical" path
#1338 opened Sep 12, 2019 by schmrlng
3
jax.numpy functions out argument fails question Questions for the JAX team
#1328 opened Sep 9, 2019 by mgbukov
3
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
Provide filename and line number for the compiling function in compilation logging enhancement New feature or request
#1247 opened Aug 26, 2019 by zhongwen
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
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
Feature request: extracting intermediate values from custom vjp enhancement New feature or request
#1197 opened Aug 16, 2019 by gehring
1
[question] Call overhead / disable parallelism
#1169 opened Aug 12, 2019 by aseyboldt
4
is keyword is not preserved under vmap. documentation
#1103 opened Aug 3, 2019 by sschoenholz
3
The cost of index_update
#1034 opened Jul 19, 2019 by samuela
3
feature request: sparse jacobian and sparse hessians enhancement New feature or request
#1032 opened Jul 18, 2019 by martinResearch
23
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
Improved error messages and documentation regarding what can/cannot be jited documentation
#953 opened Jul 1, 2019 by samuela
@mattjj
7
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
Feature Request: Add a preduce_sum method enhancement New feature or request
#839 opened Jun 11, 2019 by chaserileyroberts
ProTip! no:milestone will show everything without a milestone.