Skip to content

Pull requests: jax-ml/jax

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

Update remat config name in docs
#26929 opened Mar 5, 2025 by danijar Loading…
❤️ 1
garbage in prettyprint
#32042 opened Sep 23, 2025 by mattjj • Draft
docs: Add note about computation perturbation in debug.print/breakpoint
#26682 opened Feb 22, 2025 by alhridoy Loading…
1
8
Add shardy_mpmd extensions to jaxlib
#31661 opened Sep 9, 2025 by copybara-service bot Loading…
[Mosaic:TPU][NFC] Simplify computeSqueezeDimsChecked
#31652 opened Sep 9, 2025 by copybara-service bot Loading…
[pallas:mgpu] emit_pipeline_emit_pipeline to support different number of stages for each input.
#31543 opened Sep 4, 2025 by copybara-service bot Loading…
Integrate Triton up to [7decd434](https://github.com/openai/triton/commits/7decd434a2dbf76e74ea3c2a5ec877326b83f4fa)
#31554 opened Sep 4, 2025 by copybara-service bot Loading…
feat(gh-13291): Add Pareto distribution functions: cdf, logcdf, logsf, sf, ppf, logppf, and corresponding tests
#26556 opened Feb 15, 2025 by Qazalbash Loading…
Add sample_image.
#30872 opened Aug 9, 2025 by carlosgmartin Loading…
4
Integrate Triton up to
#31739 opened Sep 11, 2025 by copybara-service bot Loading…
Fix Jax OSS CI CI Optional GPU Presubmit Label to flag PR to run additional GPU testing not in standard presubmits
#31562 opened Sep 4, 2025 by copybara-service bot Loading…
Specify init step size for odeint cla: yes
#2604 opened Apr 4, 2020 by jacobjinkelly • Draft
6
Set default odeint tol appropriately for float32 cla: yes
#3444 opened Jun 15, 2020 by shoyer Loading…
1
2
Support building with Bzlmod
#32029 opened Sep 23, 2025 by meteorcloudy • Draft
2
Unset KUBERNETES_SERVICE_HOST environment variable
#28363 opened Apr 28, 2025 by mwhittaker • Draft
jax profiler - trace events with absolute timestamps #26288
#26873 opened Mar 1, 2025 by Utkarsh-Aggarwal Loading…
1
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…
[Bugfix] Ensure consistent output of jax.lax.map via fixed-size chunking (#27591)
#27592 opened Mar 29, 2025 by SwarnimShekhar Loading…
1
improve raised error in conv_general_dilated
#22607 opened Jul 23, 2024 by NDOWAH Loading…
[Pallas/TPU] Add flash attention math tutorial
#20145 opened Mar 8, 2024 by sharadmv • Draft
1
Remove broken jax/jaxlib version test. pull ready Ready for copybara import and testing
#28515 opened May 5, 2025 by hawkinsp Loading…
Debugging CI failure.
#28035 opened Apr 15, 2025 by hawkinsp • Draft
Test PR for runner
#30315 opened Jul 18, 2025 by quoctruong • Draft
[DOC] Add documentation for jax_debug_infs configuration option to JAX debugging flags
#27012 opened Mar 8, 2025 by sumana-2705 Loading…
2
[Mosaic:TPU] Better tpu.memref_reshape support for packed types by improving the reshaped layout inference
#31418 opened Aug 28, 2025 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.