ROCm/jax

Pull Requests Commits

Merge pull request #4913 from jblespiau:changelist/342345375

a-googler committed 5 years ago

28d4f4c5

Fix float indices in lax.slice

gnecula committed 5 years ago

64022922

[jax2tf] Add tests for the conversion of concatenate.

bchetioui committed 5 years ago

b12b39fb

Merge pull request #4925 from bchetioui:test_select

a-googler committed 5 years ago

44e671fa

Disable the remaining failing tests, remove jax2tf documentation

gnecula committed 5 years ago

3bf3e352

[jax2tf] Add testing for select_p conversion.

bchetioui committed 5 years ago

81770e82

fix grad-of-switch test under x64 mode

froystig committed 5 years ago

76d43035

change tree+aval check to use f-strings

froystig committed 5 years ago

dc1f4b8c

fix capitalization

mattjj committed 5 years ago

Verified f180ea86

update jax repository citation

mattjj committed 5 years ago

Verified c2a98c18

Merge pull request #4906 from bchetioui:test_clamp

a-googler committed 5 years ago

59e92d31

in grad-of-switch test, avoid relying on unary operations to remove weak types

froystig committed 5 years ago

7bf9a239

Merge pull request #4914 from jakevdp:fix-is-weak

a-googler committed 5 years ago

71b36972

Merge pull request #4894 from LenaMartens:changelist/342264327

a-googler committed 5 years ago

e56c3f43

Fix dtypes.is_weakly_typed

jakevdp committed 5 years ago

7f49d4ed

Change the DeviceArray error from ValueError to RuntimeError when deleted.

jblespiau committed 5 years ago

5c8286f9

Merge pull request #4907 from hawkinsp:c128scatter

a-googler committed 5 years ago

7fc5ad7e

Merge pull request #4908 from bchetioui:fix_flake_py3.9

a-googler committed 5 years ago

ed5f5133

Merge pull request #4910 from hawkinsp:xla

a-googler committed 5 years ago

c22a0667

hawkinsp committed 5 years ago

1e7bf554

Merge pull request #4896 from google:issue4696

a-googler committed 5 years ago

ac9d862a

[jax2tf] Fix mypy error in harnesses.

bchetioui committed 5 years ago

4b6301a1

Lower complex128 scatter-add to two float64 scatter-adds on GPU.

hawkinsp committed 5 years ago

b1729840

[jax2tf] Fix conversion of clamp and add testing code.

bchetioui committed 5 years ago

36e107eb

Merge pull request #4874 from bchetioui:test_broadcast

a-googler committed 5 years ago

0da1fbe2

test grad-of-switch with weak type mismatch

froystig committed 5 years ago

ca656ece

check that jaxpr types match in cond transpose using `core.typematch` rather than `__eq__`

froystig committed 5 years ago

156024fb

fix check_jaxpr docstring

froystig committed 5 years ago

78c6e4e5

adapt masking rev tests for soundness

mattjj committed 5 years ago

9a0207ab

Merge pull request #4803 from AdrienCorenflos:patch-1

a-googler committed 5 years ago

65a7f608

Newer Older