ROCm/jax

Pull Requests Commits

Merge pull request #18928 from jakevdp:jnp-linalg

a-googler committed 2 years ago

1851447b

array api: add jnp.linalg.cross & jnp.linalg.outer

jakevdp committed 2 years ago

fe2ad892

Merge pull request #18943 from jakevdp:xla-bridge

a-googler committed 2 years ago

2fa90e1d

Merge pull request #18888 from superbobry:pp-improvement

a-googler committed 2 years ago

616f4d29

Add missing __future__ import

jakevdp committed 2 years ago

796449e3

Increase test timeout for Cloud TPU nightly CI.

Jieying Luo committed 2 years ago

8128b54e

`mesh_utils.create_hybrid_device_mesh`: make sorting granules by key user configurable.

a-googler committed 2 years ago

94d58b72

The pretty printer now de-duplicates identical jaxprs

superbobry committed 2 years ago

840abfb7

[export] Add support for serialization and deserialization of Exported

gnecula committed 2 years ago

b077483b

Internal only changes.

majnemer committed 2 years ago

4347950d

Update XLA dependency to use revision

a-googler committed 2 years ago

d08d2dcd

Merge pull request #18262 from jakevdp:key-reuse-jaxpr

a-googler committed 2 years ago

b225c86f

Merge pull request #18925 from gnecula:drop_enable_xla_tests

a-googler committed 2 years ago

3b2902d0

Integrate StableHLO at openxla/stablehlo@ab709fe4

GleasonK committed 2 years ago

184e3a88

[XLA:Python] Add support for explicitly creating the gloo tcp context.

hawkinsp committed 2 years ago

384e29e3

Add experimental static key reuse checking

jakevdp committed 2 years ago

a52d1878

[XLA:Mosaic] Support tpu.bitcast for i16, i8.

bythew3i committed 2 years ago

3651d4c4

Merge pull request #18924 from jakevdp:ci-ratchet

a-googler committed 2 years ago

fc1b4b04

Remove enable_xla=False test harnesses from export test

gnecula committed 2 years ago

f4a76bd7

CI: use ratchet to pin actions commits

jakevdp committed 2 years ago

b1e9afca

Merge pull request #18923 from jakevdp:ci-setup-python

a-googler committed 2 years ago

a721ae2a

[jax2tf] Remove backwards compatibility shims (jax2tf.shape_poly)

gnecula committed 2 years ago

b5f852f7

CI: update actions/setup-python to v5

jakevdp committed 2 years ago

bdd8318e

Add unroll=True to all the calls of fori_loop

a-googler committed 2 years ago

9d0a9918

`Effects` is now an immutable set

superbobry committed 2 years ago

352e10ed

[XLA:MOSAIC] implement downcast from s32 to s8 correctly

blakehechtman committed 2 years ago

1cba270f

Use explicit test case names instead of the numbering system of parameterized

apaszke committed 2 years ago

bb0405e5

Merge pull request #18914 from apaszke:capi-fix

a-googler committed 2 years ago

f52f3a0b

Skip DLPack tests that are failing with the recent changes to the C API

apaszke committed 2 years ago

9cbf2411

Update XLA dependency to use revision

a-googler committed 2 years ago

b25e2e67

Newer Older