Skip to content

Pull requests: jax-ml/jax

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

Relax shape check in _compute_fans.
#32739 opened Oct 19, 2025 by copybara-service bot 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…
Move xla literal out of xla_compiler
#32731 opened Oct 18, 2025 by copybara-service bot Loading…
Add logger to libtpu sdk.
#32726 opened Oct 17, 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…
[Pallas:MGPU] PyTorch collectives export WIP
#32719 opened Oct 17, 2025 by copybara-service bot Loading…
Fix typo in pjit.py
#32718 opened Oct 17, 2025 by rniczh 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…
Optimize multi_weakref_lru_cache.
#32682 opened Oct 16, 2025 by copybara-service bot Loading…
Remove Traced.in_avals property
#32665 opened Oct 16, 2025 by copybara-service bot Loading…
remove literal from xla_compiler
#32662 opened Oct 16, 2025 by copybara-service bot Loading…
Reverts 1bd3402c63f3c5c0614af1d732389230d18ec237
#32650 opened Oct 15, 2025 by copybara-service bot Loading…
[mutable-arrays] enable vjp3 by default
#32635 opened Oct 15, 2025 by copybara-service bot Loading…
Use nvml impl lib based wrapper
#32633 opened Oct 15, 2025 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2025-09-19.