ROCm/jax

Pull Requests Commits

print partial eval stats in jit

mattjj committed 5 years ago

1e3f7005

Merge pull request #4376 from SIben:deterministic_limited_support_order

a-googler committed 5 years ago

b040df20

[jax2tf] Make rows of the generated documentation ordered deterministically.

bchetioui committed 5 years ago

4ba0ba6f

Merge pull request #4361 from SIben:test_eigh_p

a-googler committed 5 years ago

6d791b3e

Merge pull request #4359 from apaszke:fix-ppermute-batching

a-googler committed 5 years ago

ea754674

[jax2tf] Implementation of the conversion of eigh_p.

bchetioui committed 5 years ago

bbbb7ce2

Merge pull request #4373 from google:avoid-device-interactions

a-googler committed 5 years ago

e4d9845b

only test if omnistaging is enabled

mattjj committed 5 years ago

34e04609

Merge pull request #4331 from zhangqiaorjc:rm_fretchet

a-googler committed 5 years ago

70516c42

mattjj committed 5 years ago

6b60267e

Merge pull request #4362 from jakevdp:type-annotation

a-googler committed 5 years ago

a62763e0

Merge pull request #4371 from jakevdp:moveaxis-fix

a-googler committed 5 years ago

d43d5d90

move a _device_put_raw under broadcast impl

mattjj committed 5 years ago

2abb37c2

Remove unused var. Bump tol.

zhangqiaorjc committed 5 years ago

b6e9da36

Merge pull request #4299 from zhangqiaorjc:segsum

a-googler committed 5 years ago

04fa89a1

jnp.moveaxis: fix bug when axes are integer dtype

jakevdp committed 5 years ago

2cf8d49f

Merge pull request #4366 from jakevdp:diff-empty

a-googler committed 5 years ago

2cb795e0

Merge pull request #4347 from jakevdp:array-equiv

a-googler committed 5 years ago

ada6f30f

call _check_arraylike in jnp.diff

jakevdp committed 5 years ago

18054e05

Add ceil_of_ratio util and bucket_size TODO.

zhangqiaorjc committed 5 years ago

35d23199

Clean-up todos related to the upgrade of jaxlib.

a-googler committed 5 years ago

55c6bdfe

Bump tol of float32 for complex64 inner product.

zhangqiaorjc committed 5 years ago

83f14012

Internal change

jblespiau committed 5 years ago

9f53d2a8

Merge pull request #4345 from jakevdp:indentation

a-googler committed 5 years ago

24fe07eb

Merge pull request #4229 from johnpjf:changelist/330579231

a-googler committed 5 years ago

c33335b2

Implement jnp.array_equiv

jakevdp committed 5 years ago

ce1ce9cb

Merge branch 'master' into rm_fretchet

zhangqiaorjc committed 5 years ago

Verified e76ebea9

Change segment_sum to use no bucketing by default.

zhangqiaorjc committed 5 years ago

614acce4

Add test matrices that exercise more code path.

zhangqiaorjc committed 5 years ago

d7564c50

Test pmap/vmap interactions of all reduction collectives

apaszke committed 5 years ago

2081e5ac

Older