ROCm/jax

Pull Requests Commits

Merge pull request #24881 from dfm:ffi-call-rep-rule

Google-ML-Automation committed 1 year ago

c6051b3e

Update array-api-tests commit

jakevdp committed 1 year ago

a115b2ce

Merge pull request #24871 from carlosgmartin:numpy_put_along_axis

Google-ML-Automation committed 1 year ago

4fe91645

Merge pull request #24905 from jakevdp:old-arg

Google-ML-Automation committed 1 year ago

5764afb4

Merge pull request #24567 from Intel-tensorflow:minigoel/intel-plugin

Google-ML-Automation committed 1 year ago

8e292122

Add shard map replication rule for ffi_call.

dfm committed 1 year ago

41a0493e

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

c40d405e

cleanup: delete unused argument from internal reduction helper

jakevdp committed 1 year ago

4a3e1155

Merge pull request #24904 from jakevdp:array-api

Google-ML-Automation committed 1 year ago

04d339d6

Add numpy.put_along_axis.

carlosgmartin committed 1 year ago

1f114b1c

Merge pull request #24864 from jakevdp:logaddexp2

Google-ML-Automation committed 1 year ago

303b792a

Update array-api-tests commit

jakevdp committed 1 year ago

d0f36666

jnp.logaddexp2: simplify implementation

jakevdp committed 1 year ago

d823f172

Move Control Flow text from Sharp Bits into its own tutorial.

emilyfertig committed 1 year ago

e6f6a8af

Merge pull request #24897 from hawkinsp:ipow

Google-ML-Automation committed 1 year ago

19a51de2

Merge pull request #24896 from hawkinsp:pad

Google-ML-Automation committed 1 year ago

fcde8aa6

Merge pull request #24751 from Stella-S-Yan:feature/default_device_str

Google-ML-Automation committed 1 year ago

cea81767

[sharding_in_types] Support shard_map with sharding in types. Right now only full manual mode is supported.

yashk2810 committed 1 year ago

05716b58

[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes.

Google-ML-Automation committed 1 year ago

a8464ce7

Merge pull request #24865 from jakevdp:dep-clear-backends

Google-ML-Automation committed 1 year ago

89f411ab

Fix pgle test handling of directory removal.

dfm committed 1 year ago

de07be5c

Merge pull request #24899 from vfdev-5:patch-2

Google-ML-Automation committed 1 year ago

3380a211

finalize deprecation of jax.clear_backends

jakevdp committed 1 year ago

f401c979

Fixed typo in asan.yaml

vfdev-5 committed 1 year ago

Verified 6644bbc9

Merge pull request #23374 from jaro-sevcik:mock-topology-config

Google-ML-Automation committed 1 year ago

a1eb5cea

Don't use an out-of-line lowering for integer_pow for small powers.

hawkinsp committed 1 year ago

081eaeaa

Make the jaxpr for jnp.pad in "constant" mode more succinct.

hawkinsp committed 1 year ago

ad5a0621

[pallas:mosaic_gpu] Ported two pipelining optimizations to `emit_pipeline`

superbobry committed 1 year ago

aefe6215

Merge pull request #23805 from dfm:ffi-examples-state

Google-ML-Automation committed 1 year ago

83700828

Merge pull request #24889 from skye:barrier

Google-ML-Automation committed 1 year ago

268c86f8

Newer Older