ROCm/jax

Pull Requests Commits

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

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

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

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

Merge pull request #4803 from AdrienCorenflos:patch-1

a-googler committed 5 years ago

65a7f608

Fix select_and_gather_add_transpose rule.

LenaMartens committed 5 years ago

91777c33

Merge pull request #4854 from j-towns:tidy-stack

a-googler committed 5 years ago

83a38f4f

add correct annotations to core.TraceStack

mattjj committed 5 years ago

8b006f6a

Merge pull request #4886 from skye:jaxlib_fixes2

a-googler committed 5 years ago

18120020

Merge pull request #4890 from google:autodiff-cookbook-fix

a-googler committed 5 years ago

8358c016

fix typo in autodiff cookbook

mattjj committed 5 years ago

491ad965

Merge pull request #4887 from skye:jaxlib

a-googler committed 5 years ago

a81179f6

README etc. updates for new jaxlib release

skye committed 5 years ago

36799f50

Merge pull request #4832 from shoyer:gmres_qr

a-googler committed 5 years ago

66cf93f1

Merge pull request #4877 from jakevdp:prng-seed-invariance

a-googler committed 5 years ago

ce360e69

Relax GMRES test tolerances

shoyer committed 5 years ago

fca56663

jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support

skye committed 5 years ago

5c89170d

Older