Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
469 Open 15,036 Closed
469 Open 15,036 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

Adding the elliptical integral functions ellipj, ellipk, ellipe. cla: yes
#2492 opened Mar 23, 2020 by botev Loading…
@mattjj
18
Add sample_image.
#30872 opened Aug 9, 2025 by carlosgmartin Loading…
4
Adds warning, in particular to catch Partial(Partial(...), ...)
#15999 opened May 14, 2023 by patrick-kidger Loading…
1
@hawkinsp
[TSAN CI] Try NO_CYTHON_COMPILE=true when installing Cython
#28967 opened May 23, 2025 by vfdev-5 • Draft
Replace [with_cuda] by [with-cuda] in setup.py
#27317 opened Mar 21, 2025 by paLeziart Loading…
@jakevdp
3
[CI] Introduce action lint workflow
#29355 opened Jun 10, 2025 by MichaelHudgins • Draft
[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…
Propagate NCCL aborts into user exceptions.
#30810 opened Aug 6, 2025 by copybara-service bot Loading…
@mwhittaker
Testing TPU CI (B)
#31137 opened Aug 18, 2025 by copybara-service bot Loading…
Make //jax/experimental:host_callback an alias for //jax/experimental.
#30899 opened Aug 11, 2025 by copybara-service bot Loading…
@jakevdp
Update jax.grad docs
#24924 opened Nov 15, 2024 by 8bitmp3 • Draft
Uses include to reference xla packages in jaxlib xla BUILD
#28304 opened Apr 25, 2025 by yliu120 Loading…
Test PR for runner
#30315 opened Jul 18, 2025 by quoctruong • Draft
[mutable-arrays] systematic scan ad test
#29834 opened Jun 28, 2025 by mattjj • Draft
[pallas:mosaic_gpu] Sketched a decorator exposing Pallas Mosaic GPU kernels to PyTorch
#31663 opened Sep 9, 2025 by copybara-service bot Loading…
@superbobry
Ignore more XLA flags for the JAX cache.
#22624 opened Jul 24, 2024 by nouiz Loading…
1
[DOC] Add documentation for jax_debug_infs configuration option to JAX debugging flags
#27012 opened Mar 8, 2025 by sumana-2705 Loading…
2
Update jax.numpy docs
#24920 opened Nov 15, 2024 by 8bitmp3 • Draft
[WIP] skip zeroing weights when computing weight matrix for scale_and_translate
#24111 opened Oct 4, 2024 by barney-s Loading…
@dfm
Prefer user specified dtype in jax.nn.initializers.orthogonal
#18266 opened Oct 24, 2023 by ethanluoyc Loading…
1
@jakevdp
30
[WIP] Add a custom_ad interface to allow simultaneous customization of JVP and VJP rules
#22457 opened Jul 15, 2024 by dfm • Draft
@dfm
Update JAX Ecosystem
#23163 opened Aug 21, 2024 by 8bitmp3 Loading…
@jakevdp
Fixing broken fused_attention_stablehlo_test_gpu.
#28559 opened May 6, 2025 by mwhittaker Loading…
@mwhittaker
1
try giving outside_call_p in/outfeed effects, tests fail
#19968 opened Feb 24, 2024 by mattjj • Draft
Add Lineax link to jnp.linalg.solve
#17310 opened Aug 26, 2023 by patrick-kidger • Draft
ProTip! Exclude everything labeled bug with -label:bug.