ROCm/jax

Pull Requests Commits

Merge pull request #20771 from rajasekharporeddy:doc_typos

a-googler committed 1 year ago

06cd05d1

[array api] update to latest test repo commit

jakevdp committed 1 year ago

572c1628

DOC: Fix docstring typos in scipy special functions

rajasekharporeddy committed 1 year ago

660d612e

Update XLA dependency to use revision

a-googler committed 1 year ago

63a9ff3b

relax a side-effects test that was erroneously checking against a canonical order

mattjj committed 1 year ago

192e2b6c

Fix minor typo in Pallas docs.

justinjfu committed 1 year ago

f8919a32

Merge pull request #20755 from Micky774:array-api-unstack

a-googler committed 1 year ago

cd9dcd20

Add new unstack function to numpy/array_api namespaces

Micky774 committed 1 year ago

6bdc83c6

Async dispatch expensive computations on the JAX CPU backend.

yueshengys committed 1 year ago

64775d02

Add layout support to `make_array_from_callback`.

yashk2810 committed 1 year ago

eb92a5c7

Merge pull request #20734 from hawkinsp:atfork

a-googler committed 1 year ago

b9a853d0

Merge pull request #20754 from Micky774:array-api-hypot

a-googler committed 1 year ago

5f22b125

Merge pull request #20703 from jakevdp:metal-ci-nightly

a-googler committed 1 year ago

eabd1bc7

Merge pull request #20753 from Micky774:array-api-expose

a-googler committed 1 year ago

982ab533

Fixed hypot bug on nan/inf pairings, began deprecation of non-real values

Micky774 committed 1 year ago

2899213e

fix matrix dimension and block shape.

a-googler committed 1 year ago

51352fa0

Expose existing functions in array API namespace

Micky774 committed 1 year ago

8b93da18

Accept layout on `ShapeDtypeStruct` on the `sharding` argument. `DeviceLocalLayout.AUTO` is not allowed on SDS.

yashk2810 committed 1 year ago

90401d51

Merge pull request #20762 from j-towns:scatter-doc-correction

a-googler committed 1 year ago

b7293005

Optimize `_create_copy_plan` by iterating over only the shards that are needed for materialization

junwhanahn committed 1 year ago

ac1a53d8

Merge pull request #20586 from superbobry:jaxlib

a-googler committed 1 year ago

3a094044

Correct a name in ScatterDimensionNumbers docstring

j-towns committed 1 year ago

b2783120

Update XLA dependency to use revision

a-googler committed 1 year ago

78c056f4

If callback returns a fully replicated global array, return it as is.

yashk2810 committed 1 year ago

2c85ca6f

[XLA] Clear derived instruction's sharding only if shapes are incompatible.

a-googler committed 1 year ago

4a6ee78f

Update XLA dependency to use revision

a-googler committed 1 year ago

7d0ba76e

Add support for loading checkpoints with a given layout to the array serialization library

yashk2810 committed 1 year ago

5ce7dca9

Bumped the minimum jaxlib to 0.4.23

superbobry committed 1 year ago

754fab91

Remove the dead code now that jax.Array is the only array we have

yashk2810 committed 1 year ago

70dca303

Update XLA dependency to use revision

a-googler committed 1 year ago

ee8ce0f8

Newer Older