ROCm/jax

Pull Requests Commits

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

b4eb48cd

Merge pull request #28664 from mattjj:rahul-fix

Google-ML-Automation committed 1 year ago

ddfcf84a

Automated Code Change

Google-ML-Automation committed 1 year ago

2016e596

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

6ec18d49

[si_vjp] fix bugs around symbolic zeros

mattjj committed 1 year ago

55a9de3c

[Pallas] Allow more int casting tests.

WindQAQ committed 1 year ago

35e2657b

Fix typo in FusedAttentionTest

justinjfu committed 1 year ago

0aef4475

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

yashk2810 committed 1 year ago

4aee3810

Merge pull request #28626 from jakevdp:rm-div

Google-ML-Automation committed 1 year ago

7ba83d83

Merge pull request #28656 from hawkinsp:profiler

Google-ML-Automation committed 1 year ago

d4645650

Merge pull request #28650 from olupton:pytest-importlib

Google-ML-Automation committed 1 year ago

a4edfaca

[Mosaic] Fix typo: FPToSI > SIToFP.

WindQAQ committed 1 year ago

6e97afda

Disable profiler tests under Python 3.14 if multithreaded.

hawkinsp committed 1 year 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 1 year ago

0605bc4a

Reverts 8137c37e324c9cb5c8f991a16d78310b6e37bd05

emilyfertig committed 1 year ago

701af068

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

junwhanahn committed 1 year ago

d0858a2c

Merge pull request #28642 from hawkinsp:tsan

Google-ML-Automation committed 1 year ago

a4612e41

[Mosaic] Move sitofp lowering to Mosaic.

WindQAQ committed 1 year ago

9eb8c9ea

Use cython from pypi in tsan CI build.

hawkinsp committed 1 year ago

7caebde7

Add `.update` to ShapeDtypeStruct

yashk2810 committed 1 year ago

e194d532

Split custom_* tests out of api_test into new target.

dfm committed 1 year ago

d551839d

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

Google-ML-Automation committed 1 year ago

87a8051e

sharadmv committed 1 year ago

60892e69

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

9428fa04

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

apaszke committed 1 year ago

244cb362

pytest: use importlib mode by default

olupton committed 1 year ago

e48fe360

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

blakehechtman committed 1 year ago

60212a39

Add supports_pinned_allocator to allow debugging pinning issues.

pschuh committed 1 year ago

f8a3f061

Add host offloading docs to public website

yashk2810 committed 1 year ago

56bbaa36

Fix empty string handling for cloud_tpu_cluster

Google-ML-Automation committed 1 year ago

da8f62b4

Older