Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
278 Open 8,175 Closed
278 Open 8,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
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

BFGS algorithm cla: yes
#3101 by Joshuaalbert was merged Jul 29, 2020 Loading…
151
Add initial LOBPCG top-k eigenvalue solver (#3112) pull ready Ready for copybara import and testing
#10962 by vlad17 was merged Jun 30, 2022 Loading…
57
an experiment in handling instances with __jax_array__ cla: yes pull ready Ready for copybara import and testing
#4725 by mattjj was merged Dec 16, 2020 Loading…
42
Implement scipy.sparse.linalg.cg cla: yes
#1821 by TuanNguyen27 was closed Apr 7, 2020 Loading…
41
make custom_transforms handle pytrees, add api.defvjp cla: yes
#818 by mattjj was merged Jun 11, 2019 Loading…
1 task done
1
40
Issue1635 expm cla: yes
#1940 by sriharikrishna was merged Jan 22, 2020 Loading…
39
Implement np.linalg.pinv cla: yes
#1656 by TuanNguyen27 was merged Dec 3, 2019 Loading…
38
[jax2tf(js)] Convertible QuickDraw model using Linen cla: yes pull ready Ready for copybara import and testing
#4606 by bchetioui was merged Nov 5, 2020 Loading…
38
Implementation of np.linalg.{cond, tensorinv} cla: yes
#2125 by StephenHogg was merged Feb 7, 2020 Loading…
37
Initial commit of jax2iree cla: yes
#7425 by stellaraccident was closed Nov 29, 2021 • Draft
31
Add SVD to jax/numpy/linalg, jax/scipy/linalg cla: yes
#197 by vishwakftw was merged Jan 8, 2019 Loading…
30
WIP: Add bicgstab implementation cla: yes
#3796 by sunilkpai was closed Feb 22, 2021 • Draft
29
[jax2tf] implementation of select_and_gather_add conversion. cla: yes
#3987 by bchetioui was merged Aug 7, 2020 Loading…
28
Promotion via lattice cla: yes pull ready Ready for copybara import and testing
#4788 by jakevdp was merged Nov 6, 2020 Loading…
26
Issue1635 expm frechet cla: yes
#2062 by sriharikrishna was merged Jun 28, 2020 Loading…
23
Add jax.image.resize. cla: yes
#3703 by hawkinsp was merged Jul 10, 2020 Loading…
23
Changes to allow pmap-of-sharded_jit to run on multi-host platforms. cla: yes
#3237 by skye was merged May 29, 2020 Loading…
22
WIP: named call context manager cla: yes pull ready Ready for copybara import and testing
#8290 by sharadmv was closed Feb 22, 2022 • Draft
12 of 13 tasks
22
add random.loggamma and improve dirichlet & beta implementation pull ready Ready for copybara import and testing
#9906 by jakevdp was merged Mar 21, 2022 Loading…
1
21
Static kwargs for jit cla: yes
#3532 by shoyer was closed Apr 9, 2021 Loading…
5 tasks done
1
19
jet of pow using comp with exp, mul, log cla: yes
#2652 by jacobjinkelly was merged Apr 14, 2020 Loading…
18
[jax2tf] Change np.array dtype to jnp.float32 when target dtype is bfloat16. cla: yes
#3959 by bchetioui was merged Aug 13, 2020 Loading…
1
18
Implementation of np.corrcoef cla: yes
#1071 by wyjw was merged Jul 30, 2019 Loading…
17
Grouped convs cla: yes
#859 by jheek was merged Jun 18, 2019 Loading…
17
Reimplemented fixpoint computation for scan. cla: yes
#1614 by gnecula was merged Nov 6, 2019 Loading…
17
ProTip! Mix and match filters to narrow down what you’re looking for.