Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
25 Open 3,279 Closed
25 Open 3,279 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

Asynchronous dispatch doc update regarding jax.Array migration pull ready Ready for copybara import and testing
#17965 by tttttanya was merged Oct 5, 2023 Loading…
Update setup.py and CHANGELOG for jax 0.4.18 release pull ready Ready for copybara import and testing
#17989 by skye was merged Oct 6, 2023 Loading…
Update versions post jax 0.4.18 release kokoro:force-run pull ready Ready for copybara import and testing
#17994 by skye was merged Oct 7, 2023 Loading…
Improve CUDA install documenation. kokoro:force-run pull ready Ready for copybara import and testing
#17981 by hawkinsp was merged Oct 6, 2023 Loading…
1
Fix tests that mistakenly used assertRaises(..., msg=...) to match… pull ready Ready for copybara import and testing
#17979 by hawkinsp was merged Oct 6, 2023 Loading…
Fix NumPy warning when casting negative float to unsigned in tan() te… pull ready Ready for copybara import and testing
#17971 by hawkinsp was merged Oct 6, 2023 Loading…
Check dtypes in fft_p's abstract eval rule. pull ready Ready for copybara import and testing
#17978 by hawkinsp was merged Oct 6, 2023 Loading…
Fix mypy errors in flash_attention kokoro:force-run pull ready Ready for copybara import and testing
#18038 by apaszke was merged Oct 10, 2023 Loading…
change vmap api to be curried (closes #78) cla: yes
#79 by mattjj was merged Dec 11, 2018 Loading…
add rot90 and flip, adjust testOp test selection cla: yes
#77 by mattjj was merged Dec 11, 2018 Loading…
1
add travis ci support cla: yes
#111 by mattjj was merged Dec 14, 2018 Loading…
Add unit_diagonal option to lax_linalg.solve_triangular. cla: yes
#925 by hawkinsp was merged Jun 26, 2019 Loading…
Document jax.disable_jit. Add an example to jax.grad. cla: yes
#418 by hawkinsp was merged Feb 20, 2019 Loading…
add random.fold_in, update mnist_vae.py w/ loop improvements cla: yes
#369 by mattjj was merged Feb 13, 2019 Loading…
Update references to aarch64 support in the README. kokoro:force-run pull ready Ready for copybara import and testing
#18044 by hawkinsp was merged Oct 10, 2023 Loading…
Implement numpy.linalg.solve and scipy.linalg.solve. cla: yes
#167 by hawkinsp was merged Dec 22, 2018 Loading…
Add value_and_grad (closes #149) cla: yes
#151 by mattjj was merged Dec 20, 2018 Loading…
Implement the mode='constant' case of np.pad. cla: yes
#213 by hawkinsp was merged Jan 10, 2019 Loading…
gate np.ptp tests by numpy version check >= 1.15 cla: yes
#328 by mattjj was merged Feb 5, 2019 Loading…
add flag to disable jit globally (fixes #252) cla: yes
#337 by mattjj was merged Feb 7, 2019 Loading…
add batching rule for cholesky PAIR=phawkins cla: yes
#327 by mattjj was merged Feb 5, 2019 Loading…
Adding unimplemented functions more numpy modules cla: yes
#72 by alexbw was merged Dec 11, 2018 Loading…
skip some cases to satisfy internal tests cla: yes
#313 by mattjj was merged Feb 3, 2019 Loading…
improve error messages for lax.slice/index funs cla: yes
#310 by mattjj was merged Feb 3, 2019 Loading…
MAINT Drop underscore from the name of externally-referenced state objects pull ready Ready for copybara import and testing
#18111 by superbobry was merged Oct 14, 2023 Loading…
ProTip! Adding no:label will show everything without a label.