-
Notifications
You must be signed in to change notification settings - Fork 3.3k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
feat: Add weights parameter to quantile and percentile functions (#32647)
#34084
opened Dec 22, 2025 by
C-you-know
Loading…
[NFC] Remove obsolete version check in
absl_cpp_logging_test.
#34081
opened Dec 22, 2025 by
copybara-service
bot
Loading…
Add file identifier and extension to export schema
#34075
opened Dec 20, 2025 by
lutzroeder
Loading…
[dep] finalize a number of deprecations for JAX v0.9.0
#34067
opened Dec 19, 2025 by
copybara-service
bot
Loading…
Add support for specifying NCCL version with nccl_version flag
#34063
opened Dec 19, 2025 by
ezhulenev
Loading…
Fix the licenses in
jax/BUILD file as a follow-up after [this change](https://github.com/jax-ml/jax/commit/7bc428b8a0204b384e4236181ebe5bf1cac02b5c).
#34057
opened Dec 19, 2025 by
copybara-service
bot
Loading…
[pmap] Update migrate pmap docs with additional user errors and performance notes.
#34053
opened Dec 19, 2025 by
copybara-service
bot
Loading…
[Mosaic GPU] Add support for TMA gather/scatter in WG semantics.
#34052
opened Dec 19, 2025 by
copybara-service
bot
Loading…
Add programmatic profiling test with session_id support in JAX.
#34051
opened Dec 19, 2025 by
copybara-service
bot
Loading…
Update convertReshardToShardedToUnreduced to generate replicated-to-unreduced + sharded-to-unreduced.
#34038
opened Dec 19, 2025 by
copybara-service
bot
Loading…
Update the lock files post the 0.8.2 release.
#34035
opened Dec 18, 2025 by
copybara-service
bot
Loading…
[Autotuner] Combine all* autotuners in one backend-agnostic pass.
#34034
opened Dec 18, 2025 by
copybara-service
bot
Loading…
Add the targets needed for Bazel tests but not included in JAX wheels in
wheel_additives.
#34031
opened Dec 18, 2025 by
copybara-service
bot
Loading…
[Pallas/Mosaic GPU] Make Mosaic GPU the default lowering backend for Pallas on GPU.
#34027
opened Dec 18, 2025 by
copybara-service
bot
Loading…
[Docs] lax.conv_shape_tuple: Docstring and clarified shape assumptions and padding
#34026
opened Dec 18, 2025 by
Prakharprasun
Loading…
[Pallas] Add trivial batching rule for with_memory_space_constraint
#34016
opened Dec 18, 2025 by
copybara-service
bot
Loading…
Update XLA dependency to use revision http://github.com/openxla/xla/commit/f335d3d6cfc367dc32de84a76c62f2fc9a08d22e
#34008
opened Dec 17, 2025 by
copybara-service
bot
Loading…
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.