ROCm/jax

Pull Requests Commits

Fix axis_index inside nested pmaps

apaszke committed 5 years ago

332a9ba1

Merge pull request #4370 from johnpjf:patch-1

a-googler committed 5 years ago

997e56d0

Merge pull request #4376 from SIben:deterministic_limited_support_order

a-googler committed 5 years ago

b040df20

Delete dead axis_index code

apaszke committed 5 years ago

e0d1b375

[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

Adding an option to return the output tree in make_jaxpr

botev committed 5 years ago

6aa35a2a

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

Updating image_test

johnpjf committed 5 years ago

Verified ae910cdd

Make scale_and_translate take spatial dimensions

johnpjf committed 5 years ago

Verified be50847c

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

Newer Older