Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
299 Open 11,453 Closed
299 Open 11,453 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

[mosaic-gpu] add utility to get number of SMs
#29249 opened Jun 4, 2025 by Amir-19 Loading…
1
Grad of unreduced pull ready Ready for copybara import and testing
#29219 opened Jun 3, 2025 by yashk2810 Loading…
Bump fsspec from 2024.5.0 to 2025.5.1 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#29011 opened May 26, 2025 by dependabot bot Loading…
Use the builtin isinstance with MLIR types pull ready Ready for copybara import and testing
#28973 opened May 23, 2025 by superbobry Loading…
[TSAN CI] Try NO_CYTHON_COMPILE=true when installing Cython
#28967 opened May 23, 2025 by vfdev-5 • Draft
Add a linearization rule for remat
#28902 opened May 21, 2025 by dfm Loading…
2
Bump the pip group across 1 directory with 2 updates dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28835 opened May 19, 2025 by dependabot bot Loading…
1
Fix typo in persistent_compilation_cache.md
#28549 opened May 6, 2025 by jjyyxx Loading…
Remove broken jax/jaxlib version test. pull ready Ready for copybara import and testing
#28515 opened May 5, 2025 by hawkinsp Loading…
Simplifiy numpy_collate in PyTorch data loading example
#28420 opened Apr 30, 2025 by mduszyk Loading…
1
Need to raise rtol/atol for Mosaic attn bfloat16
#28404 opened Apr 29, 2025 by hanzlfs Loading…
1
Unset KUBERNETES_SERVICE_HOST environment variable
#28363 opened Apr 28, 2025 by mwhittaker • Draft
Bump importlib-resources from 6.4.0 to 6.5.2 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28352 opened Apr 28, 2025 by dependabot bot Loading…
Bump kiwisolver from 1.4.5 to 1.4.8 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#28350 opened Apr 28, 2025 by dependabot bot Loading…
Make primary sidebar open by default and toggleable.
#28313 opened Apr 27, 2025 by zhangqiaorjc Loading…
1
Uses include to reference xla packages in jaxlib xla BUILD
#28304 opened Apr 25, 2025 by yliu120 Loading…
Add debug.capture() callback wrapper for argument-free debug printing
#28284 opened Apr 25, 2025 by kcdodd Loading…
1
style: type annotate flatten_util
#28201 opened Apr 23, 2025 by nstarman Loading…
fix nan gradients in loggamma for very small parameters
#28138 opened Apr 19, 2025 by SeKim12 Loading…
Fix Dropout for mode = 'test' case in jax/example_libraries/stax.py file
#28086 opened Apr 17, 2025 by rajasekharporeddy Loading…
1
[Draft] cudnn sdpa flex attention
#28048 opened Apr 15, 2025 by Cjkkkk • Draft
Debugging CI failure.
#28035 opened Apr 15, 2025 by hawkinsp • Draft
Add jax_logging_level to common config options in docs documentation kokoro:force-run pull ready Ready for copybara import and testing
#28034 opened Apr 15, 2025 by skye Loading…
5
Fix shard_map for primitives with multiple outputs and None specs kokoro:force-run pull ready Ready for copybara import and testing
#27720 opened Apr 3, 2025 by justinjfu Loading…
13
add options to compute interpolation using a dot product and to select search algorithm
#27701 opened Apr 3, 2025 by mattbahr Loading…
1
1
ProTip! What’s not been updated in a month: updated:<2025-08-29.