Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
470 Open 14,958 Closed
470 Open 14,958 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 the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18
bug fix for 21581
#21588 opened Jun 3, 2024 by Aditya-Shandilya1182 Loading…
5
relax custom_vjp dtype checks, tweak float0 behavior pull ready Ready for copybara import and testing
#21537 opened May 31, 2024 by mattjj • Draft
add general element type experiment
#8872 opened Dec 9, 2021 by mattjj • Draft
@mattjj
x64 explorations
#9173 opened Jan 11, 2022 by mattjj • Draft
@mattjj
generalize scan weak type handling to fixpoint
#9046 opened Dec 24, 2021 by mattjj Loading…
@mattjj
add more options for jax_default_matmul_precision cla: yes
#6217 opened Mar 24, 2021 by mattjj Loading…
@mattjj
avals with names cla: yes
#4397 opened Sep 24, 2020 by jekbradbury • Draft
1
Add a prototype implementation of recursive checkpointing cla: yes
#4363 opened Sep 21, 2020 by apaszke Loading…
3 tasks
@apaszke
14
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
Fix zoom state for next iteration in line search
#11090 opened Jun 14, 2022 by david-anton Loading…
1
Documentation fix: typo in jax.lax.custom_linear_solve doc
#10836 opened May 26, 2022 by dpsimpson Loading…
Fix Cmd.use_rawinput type in debuggers
#12231 opened Sep 6, 2022 by sobolevn Loading…
@sharadmv
2
[WIP, Proposal] Improve argument annotations interface
#10746 opened May 17, 2022 by JeppeKlitgaard • Draft
3
custom transpose: nearly rewrite custom_vjp pull ready Ready for copybara import and testing
#10732 opened May 16, 2022 by froystig • Draft
@froystig
1
Run scalar integer index normalization at trace time
#10683 opened May 12, 2022 by lgeiger Loading…
1
3
change grad(jnp.abs)(0.+0.j) convention to be 1.+0j pull ready Ready for copybara import and testing
#10800 opened May 23, 2022 by mattjj Loading…
1
In some cases axis_name is a tuple instead of string, handling it in … cla: yes
#7393 opened Jul 27, 2021 by ahmedjawedaj Loading…
@apaszke
4
Fix unused variable and warnings in docs Neural Networks Tutorial cla: yes
#7333 opened Jul 20, 2021 by FelipeMartins96 Loading…
@mattjj
First prototype of explicit async timeline
#30399 opened Jul 22, 2025 by chaserileyroberts • Draft
@chaserileyroberts
normalize/rearrange (kw)args by signature in jit
#10724 opened May 16, 2022 by YouJiacheng Loading…
1
[WIP] Document argument annotations
#10677 opened May 12, 2022 by JeppeKlitgaard • Draft
Better argnums/argnames inference
#10619 opened May 7, 2022 by JeppeKlitgaard • Draft
1 task
1
@hawkinsp
9
Attempt to add Generic Wrapped
#10321 opened Apr 16, 2022 by YouJiacheng Loading…
@froystig
4
ProTip! Add no:assignee to see everything that’s not assigned.