Skip to content

Pull requests: jax-ml/jax

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

Add tagging tracer and collect/inject transforms cla: yes
#1660 by jekbradbury was closed Jan 29, 2022 Loading…
6 ❤️ 6
Add optix to experimental cla: yes
#1620 by mtthss was merged Nov 7, 2019 Loading…
9 ❤️ 3
Fix time issues in odeint reverse mode cla: yes
#2817 by samuela was merged Apr 24, 2020 Loading…
10 ❤️ 3
add tuple simplification logic cla: yes
#867 by mattjj was merged Jun 18, 2019 Loading…
❤️ 2
make jax.random default dtypes 64-bit cla: yes
#761 by mattjj was merged May 23, 2019 Loading…
1
9 ❤️ 2
make custom_jvp handle all tracers gracefully cla: yes
#4112 by mattjj was closed Oct 12, 2020 • Draft
1 ❤️ 2
Add more API set up for Mock GPU client. Also clean up previous mock GPU client
#17838 by copybara-service bot was closed Oct 2, 2023 Loading…
❤️ 2
Issue1635 expm cla: yes
#1940 by sriharikrishna was merged Jan 22, 2020 Loading…
39 ❤️ 2
Add batching rule for reduce_window_p. Allows batching of np.cumprod. cla: yes
#1015 by hawkinsp was merged Jul 13, 2019 Loading…
1
1 ❤️ 1
factor out ode tests into tests/ode_test.py cla: yes
#1367 by mattjj was closed Apr 1, 2020 Loading…
2 ❤️ 1
BFGS algorithm cla: yes
#3101 by Joshuaalbert was merged Jul 29, 2020 Loading…
151 ❤️ 1
Cache test_utils.format_shape_and_dtype_string. cla: yes
#2959 by hawkinsp was merged May 5, 2020 Loading…
❤️ 1
Curry jit, grad, value_and_grad and vmap. cla: yes
#3736 by BuddenD was closed Jul 24, 2020 Loading…
3 ❤️ 1
Add a config for using Clang on Windows.
#21082 by copybara-service bot was closed May 6, 2024 Loading…
1 ❤️ 1
Stackless [WIP] pull ready Ready for copybara import and testing
#23299 by dougalm was closed Nov 20, 2024 • Draft
1 ❤️ 1
[JAX] Add python3.14 wheel test configs CI Connection Halt - Always Flag every job that has its workflow file setup for halting to wait at the designated point
#28440 by kanglant was closed Jun 12, 2025 • Draft
❤️ 1
Bump shard_count for shard_map_test to fix timeouts.
#17956 by copybara-service bot was closed Oct 5, 2023 Loading…
1
[Mosaic] apply_vector_layout C++ rewrite (17): for.op (from cl/568376871)
#17872 by copybara-service bot was closed Oct 5, 2023 Loading…
random_test: Split into two so that each target is small enough to fit within a medium timeout.
#17944 by copybara-service bot was closed Oct 5, 2023 Loading…
1
[Mosaic] apply_vector_layout C++ rewrite: Have each rule use its own builder instead of passing a builder in the context
#17871 by copybara-service bot was closed Oct 5, 2023 Loading…
[Mosaic] infer_memref_layout C++ rewrite
#17940 by copybara-service bot was closed Oct 5, 2023 Loading…
Expose mlir.ShapePolyLoweringState
#17955 by copybara-service bot was closed Oct 5, 2023 Loading…
1
[Mosaic] apply_vector_layout C++ rewrite: various bug fixes
#17939 by copybara-service bot was closed Oct 5, 2023 Loading…
[Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)
#17942 by copybara-service bot was closed Oct 5, 2023 Loading…
fix typo in scatter_add doc
#18014 by andrinr was closed Oct 9, 2023 Loading…
1
ProTip! no:milestone will show everything without a milestone.