-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add Weighted Quantile and Percentile Support to jax.numpy
#32737
opened Oct 18, 2025 by
Aniketsy
Loading…
sdk changes remove literal from xla_compiler
#32736
opened Oct 18, 2025 by
copybara-service
bot
Loading…
Use less memory during MLIR compilation.
#32732
opened Oct 18, 2025 by
copybara-service
bot
Loading…
[jax] Migrate JAX to new XLA FFI type registration
#32722
opened Oct 17, 2025 by
copybara-service
bot
Loading…
[Mosaic] Canonicalize integer
tpu::UnpackSubelementsOp
with sign_extended=false
if it's used by pack that reduces the bitwidth.
#32721
opened Oct 17, 2025 by
copybara-service
bot
Loading…
[Pallas:MGPU] PyTorch collectives export WIP
#32719
opened Oct 17, 2025 by
copybara-service
bot
Loading…
[Pallas:MGPU] Substantially refactor the PyTorch conversion logic
#32717
opened Oct 17, 2025 by
copybara-service
bot
Loading…
Add a setter for sharding for backward compatibility
#32716
opened Oct 17, 2025 by
copybara-service
bot
Loading…
[ROCm] lax.square: use chlo.square
kokoro:force-run
pull ready
Ready for copybara import and testing
#32715
opened Oct 17, 2025 by
nurmukhametov
Loading…
[pallas:sc]
plsc.kernel
now supports dict in scratch_shapes=
#32713
opened Oct 17, 2025 by
copybara-service
bot
Loading…
Remove unused bias checking code for cuDNN
dot_product_attention
.
#32711
opened Oct 17, 2025 by
copybara-service
bot
Loading…
[pallas:sc] Run SparseCore tests on presubmit on GitHub
#32696
opened Oct 16, 2025 by
copybara-service
bot
Loading…
[NFC][Mosaic:TPU] Expose hash_value of xla::Tile in header
#32681
opened Oct 16, 2025 by
copybara-service
bot
Loading…
Add
mean
argument to jnp.std
and jnp.var
families
#32668
opened Oct 16, 2025 by
arvindsankar
Loading…
[Mosaic TPU] Support stochastic rounding on older generations.
#32664
opened Oct 16, 2025 by
copybara-service
bot
Loading…
Reverts 1bd3402c63f3c5c0614af1d732389230d18ec237
#32650
opened Oct 15, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
What’s not been updated in a month: updated:<2025-09-19.