Skip to content

Pull requests: jax-ml/jax

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

WIP: better cond batching cla: yes
#2961 opened May 5, 2020 by jekbradbury • Draft
👀 1
Checkify: add debug mode with all checks+errors thrown automatically pull ready Ready for copybara import and testing
#12597 opened Sep 30, 2022 by LenaMartens Loading…
5 tasks
👀 1
FFT Convolution cla: yes
#6343 opened Apr 5, 2021 by peterroelants Loading…
12 👀 1
Fix to make jax.scipy.special.sph_harm to be close to scipy.special.sph_harm for multiple theta or phi
#20772 opened Apr 16, 2024 by rajasekharporeddy Loading…
1
9 👀 1
Bump kiwisolver from 1.4.5 to 1.4.7 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24016 opened Sep 30, 2024 by dependabot bot Loading…
Bump importlib-resources from 6.4.0 to 6.4.5 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24018 opened Sep 30, 2024 by dependabot bot Loading…
Bump hypothesis from 6.102.4 to 6.122.3 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25354 opened Dec 9, 2024 by dependabot bot Loading…
Bump fonttools from 4.51.0 to 4.55.3 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#25512 opened Dec 16, 2024 by dependabot bot Loading…
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
Update jax.tree module docs
#24925 opened Nov 15, 2024 by 8bitmp3 • Draft
Add example[newton_method]
#19160 opened Jan 2, 2024 by linjing-lab Loading…
4
[ROCm] [ROCm] BUILD file fixes and linter run
#25129 opened Nov 26, 2024 by Ruturaj4 Loading…
1
add options to compute interpolation using a dot product and to select search algorithm
#27701 opened Apr 3, 2025 by mattbahr Loading…
1
1
chore: format markdown table in README
#22986 opened Aug 10, 2024 by glitched-w0rld Loading…
Implicitly restarted arnoldi cla: yes
#4336 opened Sep 18, 2020 by mganahl • Draft
4
hoist constants to parameters in jit-compiled functions cla: yes
#3238 opened May 29, 2020 by froystig • Draft
[mutable-arrays] linearize the state
#26176 opened Jan 29, 2025 by mattjj Loading…
Add a linearization rule for remat
#28902 opened May 21, 2025 by dfm Loading…
2
Add a workflow to run benchmarks
#25386 opened Dec 10, 2024 by juliagmt-google Loading…
1
Add an example for executing serialized function in cpp
#24413 opened Oct 20, 2024 by fenghuizhang Loading…
8
Fix Dropout for mode = 'test' case in jax/example_libraries/stax.py file
#28086 opened Apr 17, 2025 by rajasekharporeddy Loading…
1
FIX: Handle WGSplatFragLayout in cond lowering.
#30238 opened Jul 15, 2025 by copybara-service bot Loading…
Simplifiy numpy_collate in PyTorch data loading example
#28420 opened Apr 30, 2025 by mduszyk Loading…
1
Codebeaver/test 723699680 26477
#26492 opened Feb 12, 2025 by CodeBeaverAI Loading…
1
[XLA:GPU] Add JAX-based precision tests for Triton and cuBLAS
#30246 opened Jul 16, 2025 by copybara-service bot Loading…
ProTip! Exclude everything labeled bug with -label:bug.