Skip to content

Pull requests: jax-ml/jax

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

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…
Ignore more XLA flags for the JAX cache.
#22624 opened Jul 24, 2024 by nouiz Loading…
1
Update jax.grad docs
#24924 opened Nov 15, 2024 by 8bitmp3 • Draft
Testing TPU CI (B)
#31137 opened Aug 18, 2025 by copybara-service bot Loading…
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…
Uses include to reference xla packages in jaxlib xla BUILD
#28304 opened Apr 25, 2025 by yliu120 Loading…
[ROCM] Jax/Pallas unit test, limiting tile to 64 in for pallas kernel in multihead attention test.
#25043 opened Nov 21, 2024 by zahiqbal Loading…
1
Bump fsspec from 2024.5.0 to 2024.10.0 dependencies Pull requests that update a dependency file python Pull requests that update Python code
#24429 opened Oct 21, 2024 by dependabot bot Loading…
Add [query,key_value]_seq_offsets arguments to dot_product_attention
#29731 opened Jun 25, 2025 by chaserileyroberts • Draft
1
Update Jupytext version to 1.16.1 in developer.md
#21164 opened May 10, 2024 by 8bitmp3 Loading…
[CI] Introduce action lint workflow
#29355 opened Jun 10, 2025 by MichaelHudgins • Draft
[mutable-arrays] systematic scan ad test
#29834 opened Jun 28, 2025 by mattjj • Draft
fix aliased arg in async pallas docs
#25977 opened Jan 18, 2025 by arvoelke Loading…
Do not attempt checking dynamic dimensions values
#23879 opened Sep 24, 2024 by erick-xanadu Loading…
1
DOC: Update gpu_performance_tips.md
#23615 opened Sep 13, 2024 by Aiemu Loading…
ProTip! Follow long discussions with comments:>50.