Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
273 Open 8,171 Closed
273 Open 8,171 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
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
Implicitly restarted arnoldi cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft
4
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…
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12
[RFC] Add solver diagnostics for cg and bicgstab cla: yes
#7825 opened Sep 6, 2021 by PhilipVinc • Draft
3
Add generic type hints to parallel
#10147 opened Apr 5, 2022 by cgarciae Loading…
prefer operator.index over int
#10395 opened Apr 21, 2022 by YouJiacheng • Draft
13
[WIP] Document argument annotations
#10677 opened May 12, 2022 by JeppeKlitgaard • Draft
Run scalar integer index normalization at trace time
#10683 opened May 12, 2022 by lgeiger Loading…
1
3
normalize/rearrange (kw)args by signature in jit
#10724 opened May 16, 2022 by YouJiacheng Loading…
1
ProTip! Adding no:label will show everything without a label.