Skip to content

/ 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.

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

474 Open 1,060 Closed
474 Open 1,060 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels.
Projects
Filter by project
Milestones
Filter by milestone
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀
`jnp.repeat` doesn't lower to `jnp.full` without an explicit axis good first issue performance
#4099 opened Aug 19, 2020 by jlebar
Parallel broadcasting (question)
#4094 opened Aug 18, 2020 by alewis
@jakevdp
2
jax.tree_utils do not keep dict key order
#4085 opened Aug 17, 2020 by Conchylicultor
7
Performance problem
#4083 opened Aug 17, 2020 by XDongiang
@jekbradbury
2
How to accurately measure GPU computing time
#4079 opened Aug 16, 2020 by XDongiang
Use of memory profiler
#4061 opened Aug 14, 2020 by jubueche
3
random.split() should allow for passing in a desired shape
#4013 opened Aug 10, 2020 by shoyer
@mattjj
4
Please support boolean types in `conv_general_dilated` enhancement
#4012 opened Aug 10, 2020 by romanngg
5
LSTM NER model trained but cannot predict
#4003 opened Aug 10, 2020 by hminooei
Defining both custom_jvp and custom_jvp question
#3996 opened Aug 8, 2020 by thisiscam
@mattjj
4
Suggestion: expose `_jit_is_disabled` utility
#3994 opened Aug 8, 2020 by KristianHolsheimer
any suggestions on how to improve performance for gradient step with odeint?
#3993 opened Aug 7, 2020 by ibulu
24
Automatic dtype promotion in loops
#3991 opened Aug 7, 2020 by shoyer
differentiating through the odeint when using complex numbers results in type errors bug
#3986 opened Aug 7, 2020 by AnnahDo
3
Prevent statements with side effects from being re-ordered question
#3976 opened Aug 6, 2020 by dionhaefner
@gnecula
11
Very slow compile when doing stochastic variational inference on the parameters of an ODE integrator performance
#3975 opened Aug 6, 2020 by ahmadsalim
4
Cannot use static scalar arrays in comparison when using omnistaging better_errors bug
#3974 opened Aug 6, 2020 by dionhaefner
@mattjj
5
`lax.reduce_window` with window dilation can produce negative shapes
#3973 opened Aug 6, 2020 by romanngg
@SIben
3
`lax.reduce_window` shape checking rule fails
#3972 opened Aug 6, 2020 by romanngg
@SIben
improve jnp.tile lowering to broadcast/reshape enhancement
#3971 opened Aug 6, 2020 by mattjj
inside pmap, grad(lambda x: psum(loss(x))) inconsistent with jvp(lambda x: psum(loss(x)) bug
#3970 opened Aug 6, 2020 by sharadmv
@mattjj
13
Bad interaction between batch trace and custom jvp trace bug
#3964 opened Aug 5, 2020 by brianwa84
@mattjj
6
Performance regression in compilation of JITed Jacobian
#3963 opened Aug 5, 2020 by ahoenselaar
2
Using `partial_eval_jaxpr` to simulate yield points
#3936 opened Aug 2, 2020 by thisiscam
6
Implement higher order interpolation in jax's map_coordinates enhancement
#3928 opened Jul 31, 2020 by fischcheng
2
ProTip! Exclude everything labeled bug with -label:bug.
You can’t perform that action at this time.