Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
546 Open 21,653 Closed
546 Open 21,653 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

Prefix PRNG Part 2: implement 1D prefix PRNG cla: yes
#2181 opened Feb 6, 2020 by j-towns Loading… updated Jun 17, 2021
17
HMM EM example cla: yes
#2405 opened Mar 12, 2020 by mattjj • Draft updated Jun 17, 2021
Handle degenerate eigenvalues in JVP rule for eigh cla: yes
#1665 opened Nov 11, 2019 by shoyer • Draft updated Jun 17, 2021
1
11
hoist constants to parameters in jit-compiled functions cla: yes
#3238 opened May 29, 2020 by froystig • Draft updated Jun 17, 2021
BDF solver for stiff ODEs cla: yes
#3781 opened Jul 16, 2020 by skrsna Loading… updated Jun 17, 2021
1 of 4 tasks
9
Set default odeint tol appropriately for float32 cla: yes
#3444 opened Jun 15, 2020 by shoyer Loading… updated Jun 17, 2021
1
2
Specify init step size for odeint cla: yes
#2604 opened Apr 4, 2020 by jacobjinkelly • Draft updated Jun 17, 2021
6
Fix init step sizes to account for the time span of integration cla: yes
#3442 opened Jun 15, 2020 by samuela Loading… updated Jun 17, 2021
1
@shoyer
11
Support mask of broadcast, consts (arange, zeros, ...) cla: yes
#3472 opened Jun 17, 2020 by juliuskunze Loading… updated Jun 17, 2021
Adds kwargable decorator cla: yes
#3461 opened Jun 16, 2020 by BuddenD Loading… updated Jun 17, 2021
4
WIP: better cond batching cla: yes
#2961 opened May 5, 2020 by jekbradbury • Draft updated Jun 17, 2021
skip fori_loops with zero trip count cla: yes
#3295 opened Jun 2, 2020 by mattjj Loading… updated Jun 17, 2021
1
4
Implicitly restarted arnoldi cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft updated Jun 17, 2021
4
Support mask(pmap) cla: yes
#4046 opened Aug 13, 2020 by j-towns • Draft updated Jun 17, 2021
@mattjj
7
Add NumPy backend cla: yes
#3923 opened Jul 31, 2020 by juliuskunze Loading… updated Jun 17, 2021
@mattjj
54
Clarify error message in scan transpose cla: yes
#3917 opened Jul 30, 2020 by jekbradbury Loading… updated Jun 17, 2021
2
avals with names cla: yes
#4397 opened Sep 24, 2020 by jekbradbury • Draft updated Jun 17, 2021
1
Add support for creating float0 arrays. cla: yes
#4437 opened Oct 2, 2020 by tomhennigan Loading… updated Jun 17, 2021
1
6
Broken custom_root + GMRES tests cla: yes
#5321 opened Jan 6, 2021 by shoyer Loading… updated Jun 17, 2021
implicitly restarted lanczos method cla: yes
#4675 opened Oct 22, 2020 by mganahl Loading… updated Jun 17, 2021
3
add pargmin_p / pargmax_p primitives cla: yes pull ready Ready for copybara import and testing
#5694 opened Feb 10, 2021 by mattjj Loading… updated Jun 17, 2021
@mattjj
13
add more options for jax_default_matmul_precision cla: yes
#6217 opened Mar 24, 2021 by mattjj Loading… updated Jun 17, 2021
@mattjj
Delay formatting of the name stack in translation rules cla: yes pull ready Ready for copybara import and testing
#6889 opened Jun 3, 2021 by apaszke Loading… updated Jun 17, 2021
@apaszke
10
Remat extensions for scan-over-layers cla: yes pull ready Ready for copybara import and testing
#6639 opened May 4, 2021 by jekbradbury • Draft updated Jun 17, 2021
@jekbradbury
1
Add a prototype implementation of recursive checkpointing cla: yes
#4363 opened Sep 21, 2020 by apaszke Loading… updated Jun 23, 2021
3 tasks
@apaszke
14
ProTip! Mix and match filters to narrow down what you’re looking for.