Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
470 Open 15,137 Closed
470 Open 15,137 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
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Adding value_and_jac* functions cla: yes P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#762 opened May 23, 2019 by matt-graham Loading…
23
optimize scan with constant-across-time outputs cla: yes
#1447 opened Oct 5, 2019 by jekbradbury • Draft
1
Handle degenerate eigenvalues in JVP rule for eigh cla: yes
#1665 opened Nov 11, 2019 by shoyer • Draft
1
11
Prefix PRNG Part 2: implement 1D prefix PRNG cla: yes
#2181 opened Feb 6, 2020 by j-towns Loading…
17
HMM EM example cla: yes
#2405 opened Mar 12, 2020 by mattjj • Draft
Adding the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18
Specify init step size for odeint cla: yes
#2604 opened Apr 4, 2020 by jacobjinkelly • Draft
6
WIP: better cond batching cla: yes
#2961 opened May 5, 2020 by jekbradbury • Draft
hoist constants to parameters in jit-compiled functions cla: yes
#3238 opened May 29, 2020 by froystig • Draft
skip fori_loops with zero trip count cla: yes
#3295 opened Jun 2, 2020 by mattjj Loading…
1
4
Set default odeint tol appropriately for float32 cla: yes
#3444 opened Jun 15, 2020 by shoyer Loading…
1
2
Adds kwargable decorator cla: yes
#3461 opened Jun 16, 2020 by BuddenD Loading…
4
Support mask of broadcast, consts (arange, zeros, ...) cla: yes
#3472 opened Jun 17, 2020 by juliuskunze Loading…
BDF solver for stiff ODEs cla: yes
#3781 opened Jul 16, 2020 by skrsna Loading…
1 of 4 tasks
9
Support mask(pmap) cla: yes
#4046 opened Aug 13, 2020 by j-towns • Draft
@mattjj
7
Implicitly restarted arnoldi cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft
4
Add a prototype implementation of recursive checkpointing cla: yes
#4363 opened Sep 21, 2020 by apaszke Loading…
3 tasks
@apaszke
14
avals with names cla: yes
#4397 opened Sep 24, 2020 by jekbradbury • Draft
1
Add support for creating float0 arrays. cla: yes
#4437 opened Oct 2, 2020 by tomhennigan Loading…
1
6
implicitly restarted lanczos method cla: yes
#4675 opened Oct 22, 2020 by mganahl Loading…
3
Broken custom_root + GMRES tests cla: yes
#5321 opened Jan 6, 2021 by shoyer Loading…
Adding value_and_jacrev and value_and_jacfwd cla: no
#5377 opened Jan 12, 2021 by tomcharnock Loading…
@mattjj
10
add pargmin_p / pargmax_p primitives cla: yes pull ready Ready for copybara import and testing
#5694 opened Feb 10, 2021 by mattjj Loading…
@mattjj
13
add more options for jax_default_matmul_precision cla: yes
#6217 opened Mar 24, 2021 by mattjj Loading…
@mattjj
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by