ROCm/jax

Pull Requests Commits

Update XLA dependency to use revision

Google-ML-Automation committed 227 days ago

b4eb48cd

Merge pull request #28664 from mattjj:rahul-fix

Google-ML-Automation committed 228 days ago

ddfcf84a

Automated Code Change

Google-ML-Automation committed 228 days ago

2016e596

Update XLA dependency to use revision

Google-ML-Automation committed 228 days ago

6ec18d49

[si_vjp] fix bugs around symbolic zeros

mattjj committed 228 days ago

55a9de3c

[Pallas] Allow more int casting tests.

WindQAQ committed 229 days ago

35e2657b

Fix typo in FusedAttentionTest

justinjfu committed 229 days ago

0aef4475

Disallow unreduced inputs for all primitives except those that implement the unreduced rule. Currently that's only `add`.

yashk2810 committed 229 days ago

4aee3810

Merge pull request #28626 from jakevdp:rm-div

Google-ML-Automation committed 229 days ago

7ba83d83

Merge pull request #28656 from hawkinsp:profiler

Google-ML-Automation committed 229 days ago

d4645650

Merge pull request #28650 from olupton:pytest-importlib

Google-ML-Automation committed 229 days ago

a4edfaca

[Mosaic] Fix typo: FPToSI > SIToFP.

WindQAQ committed 229 days ago

6e97afda

Disable profiler tests under Python 3.14 if multithreaded.

hawkinsp committed 229 days ago

0d5771cc

Make `jax.ShapeDtypeStruct` immutable. It was always supposed to be immutable inside `jax.Array` is immutable and `ShapeDtypeStruct` is a duck of `jax.Array` but immutability was never enforced.

yashk2810 committed 229 days ago

0605bc4a

Reverts 8137c37e324c9cb5c8f991a16d78310b6e37bd05

emilyfertig committed 229 days ago

701af068

Use `ValueRef` instead of `tsl::RCReference<Value>`

junwhanahn committed 229 days ago

d0858a2c

Merge pull request #28642 from hawkinsp:tsan

Google-ML-Automation committed 229 days ago

a4612e41

[Mosaic] Move sitofp lowering to Mosaic.

WindQAQ committed 229 days ago

9eb8c9ea

Use cython from pypi in tsan CI build.

hawkinsp committed 229 days ago

7caebde7

Add `.update` to ShapeDtypeStruct

yashk2810 committed 229 days ago

e194d532

Split custom_* tests out of api_test into new target.

dfm committed 229 days ago

d551839d

Merge pull request #28608 from dfm:pp-custom-jvp

Google-ML-Automation committed 229 days ago

87a8051e

sharadmv committed 229 days ago

60892e69

Update XLA dependency to use revision

Google-ML-Automation committed 229 days ago

9428fa04

[Mosaic GPU] Use explicit load/store methods instead of __getitem__/__setitem__

apaszke committed 229 days ago

244cb362

pytest: use importlib mode by default

olupton committed 229 days ago

e48fe360

[JAX:ANN] Make a test more reliable by increasing the sample size

blakehechtman committed 230 days ago

60212a39

Add supports_pinned_allocator to allow debugging pinning issues.

pschuh committed 230 days ago

f8a3f061

Add host offloading docs to public website

yashk2810 committed 230 days ago

56bbaa36

Fix empty string handling for cloud_tpu_cluster

Google-ML-Automation committed 230 days ago

da8f62b4

Older