Morph MORPH
SIGN IN SIGN UP
dhruvhead / jax UNCLAIMED
0 0 0 Python

COMMITS

main
August 30, 2026
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/496bd4bd49db9ecbffd85da630b49c860b724604
jax authors committed
5832e86
August 29, 2026
Y
Use program_order_p even if exclude_argnames is None
Yash Katariya committed
8a92dd9
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/69325769ed5c4191b16e7f5fd37d5d5ea114a18d
jax authors committed
eabf536
D
Reverts 2610691a3fcfc16d5fd09e714f492f518978622c
Dirk Hornung committed
a2f2d71
August 28, 2026
C
Explicitly propagate and specify strip_memory_space in state discharge callers.
Cristian Garcia committed
fefdef2
C
Support mutable operations (`set`, `swap`, `addupdate`) on `ReshapeTransform` and `BitcastTransform` views during state discharge.
Cristian Garcia committed
fd3802e
R
[pallas:fuser] Support full-dimension (dynamic_)slice and (dynamic_)update_slice
Robert Dyro committed
05fcbe4
J
Merge pull request #40306 from mattjj:push-unytxtzpuxsr
jax authors committed
264952c
Y
Add `exclude_argnames` on program_order.
Yash Katariya committed
94b575c
J
Merge pull request #40305 from hawkinsp:py315
jax authors committed
d551c34
M
[docs] more tweaks to complex number description
Matthew Johnson committed
c6bfad7
P
Run wheel tests using the git commit the wheels were built from.
Peter Hawkins committed
b2e4a13
J
Merge pull request #40231 from VaggelisGian:fix-intersect-setxor-size0
jax authors committed
b94bd6c
P
Use a rules_python manifest instead of patching rules_python for 3.15.
Peter Hawkins committed
75b9bd2
J
Merge pull request #40295 from hawkinsp:py315
jax authors committed
01293a5
C
Move with_memory_space_constraint primitive to jax._src.state.primitives.
Cristian Garcia committed
ab1e448
J
Merge pull request #40299 from hawkinsp:deps
jax authors committed
00894c5
Y
Dedup prev_outs with cur_inps so we don't barrier on the same input twice. In other words, if some prev_outs are in cur_inps, then skip those prev_outs from opt_barrier (since they are already covered by cur_inps).
Yash Katariya committed
ae115c2
P
Drain garbage collector before reference cycle tests.
Peter Hawkins committed
1214954
P
[Pallas/GPU interpreter] Relax test tolerances and use NumPy matmul for reference.
Peter Hawkins committed
b277e8e
P
Update tensorflow and xprof constraints in collect-profile-requirements.txt
Peter Hawkins committed
067ddf0
P
Exclude Windows Python 3.15t from the pytest build matrix.
Peter Hawkins committed
7d2846c
P
Support Python 3.15 free-threaded in Bazel builds
Peter Hawkins committed
d22ae39
J
Merge pull request #40021 from edluyuan:fix/mosaic-splat-tiled-alignment
jax authors committed
70f6ee4
P
Reland making committed an immutable field of arrays.
Peter Hawkins committed
17b4906
P
Tag tpu_ops_test notsan and nomsan.
Peter Hawkins committed
56ad652
J
Add a small cache for `xla::HloSharding` to `xla::ifrt::HloShardingSpec`
Junwhan Ahn committed
ee6d552
Y
Fix the transitive dependency graph error
Yurii Topin committed
0eaaeb4
J
Make 'committed' an immutable field of arrays.
jax authors committed
b1a91a1
J
[NCCL] Upgrade XLA NCCL version to 2.30.7
jax authors committed
c7792ef