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

Feature Request: Add a preduce_sum method enhancement New feature or request
#839 opened Jun 11, 2019 by chaserileyroberts
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 np.pad where mode is a function contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. 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
#1649 opened Nov 8, 2019 by jamieas
Expand scan to take in_axes and out_axes parameters enhancement New feature or request
#2509 opened Mar 25, 2020 by gnecula
1
@gnecula
Higher-order derivatives for determinant for low rank matrices
#2821 opened Apr 24, 2020 by dpfau
Inconsistent error messages when passing bad arguments to primitives better_errors Improve the error reporting
#2904 opened Apr 30, 2020 by gnecula
Unified API for "map" operations enhancement New feature or request
#2939 opened May 3, 2020 by shoyer
test_cg_against_scipy_shape with preconditioner=random fails with a large mismatch with a different seed bug Something isn't working
#2951 opened May 4, 2020 by hawkinsp
@shoyer
Add support for asynchronous infeed enhancement New feature or request
#3196 opened May 24, 2020 by gnecula
@gnecula
Buffer donation optimizations in AD enhancement New feature or request
#3286 opened Jun 2, 2020 by jekbradbury
Individual hessian contributions of params used in a scanned function question Questions for the JAX team
#3353 opened Jun 7, 2020 by Victorlouisdg
Unexpected overhead when using grad(pmap) performance make things lean and fast
#3612 opened Jun 30, 2020 by izmailovpavel
Support input dilation with SAME padding in lax.conv_general_dilated
#4216 opened Sep 8, 2020 by romanngg
Hacky approach to carry argument of an invertible transformation in the backward pass
#4324 opened Sep 17, 2020 by BohdanK-1QBit
Use D2D transfers for jax.jit(f)(ShardedDeviceArray).
#4341 opened Sep 18, 2020 by trevorcai
Calculating gradients only for the selected parameters
#4752 opened Nov 1, 2020 by cagrikymk
Accelerating jacobian with static_argnums question Questions for the JAX team
#4937 opened Nov 17, 2020 by gugar20
jax cpp interface, xla fixed dimensions, runtime
#5144 opened Dec 9, 2020 by lkskstlr
Mention cc compiler requirements in doc
#5384 opened Jan 12, 2021 by zhangqiaorjc
@zhangqiaorjc
remat of relu avoids rematerializing an all-zero array
#5458 opened Jan 18, 2021 by apaszke
@apaszke
Grad of custom_root fails when using cg in tangent_solve.
#5552 opened Jan 29, 2021 by MaxiLechner
@mattjj
pmap with out_axes != 0 errors when within jit
#5756 opened Feb 17, 2021 by mhauru
@jekbradbury
Support symbolic zero operations in custom_jvp enhancement New feature or request P2 (eventual) This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#5913 opened Mar 3, 2021 by srvasude
@froystig
Update MAML & TD(0) discussion in JAX 101 notebook documentation enhancement New feature or request
#5935 opened Mar 4, 2021 by jakevdp
@mattjj
Behaviour of pmap is unclear when static_broadcasted_argnums is combined with in_axes enhancement New feature or request
#6302 opened Mar 31, 2021 by alewis
@jekbradbury
ProTip! Follow long discussions with comments:>50.