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

COMMITS

main
August 28, 2026
D
Temporarily point CUDA 12.1 test container to transient image.
Dirk Hornung committed
2610691
V
numpy: return empty results for intersect1d and setxor1d at size 0
Vaggelis committed
5c671a4
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/94eb25f05a2dc944e3ff14b5c5ed9113d02870de
jax authors committed
64b2236
I
Enable DMA on boolean tensors.
Ivy Zheng committed
f078326
August 27, 2026
J
Merge pull request #40229 from VaggelisGian:fix-unique-empty-axis-shape
jax authors committed
c46889b
Y
[Mosaic TPU] Add lowering rule for `lax.approx_max_k` in Pallas Mosaic. First reduce the reduction dimension to some number of bins, then do top_k.
Yue Sheng committed
2d2fb43
P
Make 'committed' an immutable field of arrays.
Peter Hawkins committed
bbbfdcd
J
Merge pull request #39097 from teddytennant:fix-linalg-cond-singular
jax authors committed
a225a49
P
Mark several fields of PyArray as const.
Peter Hawkins committed
23dbabf
P
Remove deprecated jax.experimental.shard_map import from jax.experimental.checkify.
Peter Hawkins committed
70dde66
J
Merge pull request #40000 from Intel-tensorflow:oneapi_plugin_solver2
jax authors committed
c33b927
V
Match numpy output shapes in jnp.unique for empty arrays
Vaggelis committed
2d4a00d
P
Improve code to convert an IFRT string arrays to numpy array.
Peter Hawkins committed
ccdf7b7
I
Allow TPU `pl.kernel` to be vmapped on nontrivial batch axes.
Ivy Zheng committed
b6143ec
J
Fix Fusible.jvp to instantiate symbolic zero tangents.
jax authors committed
35d7d10
D
Reverts 5d131506a5ffe39e6d34e07d637a1db4bbe57ba3
Dirk Hornung committed
3e8c67c
C
Strip memory spaces and synchronize aliased carry in Pallas HLO interpreter.
Cristian Garcia committed
76d08b3
P
Remove an unnecessary overwrite of the .writeable flag on a numpy array.
Peter Hawkins committed
27bdcf8
P
Update the XLA pin.
Peter Hawkins committed
d5ca59f
H
[IFRT] Migrate `MemoryKind` use to use `value()` and `is_default()`
Hyeontaek Lim committed
3ce82b4
B
Reverts 683985a3e7f1877c0ffb810d87ecfbfe1e6d2614
Benjamin Chetioui committed
1f7241e
R
Propagate Mosaic GPU kernel launch errors to XLA instead of calling abort().
Rohan Jain committed
f3fcb67
D
Skip DotProductAttentionF8Test for cuDNN version 9.10.2.
Dirk Hornung committed
8f607bd
G
Remove test test_sparsecore_unsupported_gather
George Necula committed
5057f20
J
Merge pull request #40255 from hawkinsp:reqs
jax authors committed
8e1f2b7
P
Readd version constraint on matplotlib.
Peter Hawkins committed
7ea6e55
P
Make checkify error code assignment local and deterministic.
Peter Hawkins committed
8e1aa07
J
Merge pull request #40149 from edluyuan:fix-40141-eigh-large-eigenvalue-grad
jax authors committed
5fb69a3
J
Update XLA dependency to use revision http://github.com/openxla/xla/commit/cc39fef7cae080ae61fc66bfe283bb80abcf409b
jax authors committed
1da7577
C
[mgpu] Prefetch TMA descriptors during initialization.
Chris Jones committed
85ebbf6