ROCm/jax

Pull Requests Commits

Merge pull request #786 from hawkinsp/scatter

hawkinsp committed 6 years ago

Verified c65ccda3

Merge pull request #785 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 27746b8c

Add support for vmap of scatter where indices but not updates are batched.

hawkinsp committed 6 years ago

0293ecbb

Check for failure of LAPACK calls and return arrays full of NaNs if they occur.

hawkinsp committed 6 years ago

162b41e9

Updating GP regression example

JasperSnoek committed 6 years ago

3d1e419e

Updating GP regression example

JasperSnoek committed 6 years ago

58c02215

Return NaNs if Cholesky decomposition fails on CPU.

hawkinsp committed 6 years ago

600630d8

Merge pull request #783 from jekbradbury/fixes-from-pytype

mattjj committed 6 years ago

Verified 117749b7

Drop dead compiled_call code

jekbradbury committed 6 years ago

c9b2e79b

jekbradbury committed 6 years ago

9506fe6d

Merge pull request #782 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 983f9908

Fix typo in comment.

hawkinsp committed 6 years ago

fb072b7f

Improve behavior of a number of math functions for extreme inputs.

hawkinsp committed 6 years ago

6e1ec38a

Merge pull request #780 from google/issue772

mattjj committed 6 years ago

Verified d8528306

Merge pull request #781 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 65b33305

undo #503 in favor of new literal staging method

mattjj committed 6 years ago

778435a9

Enable some tests that now pass.

hawkinsp committed 6 years ago

c3add0c7

Merge pull request #771 from fehiepsi/gammaln

mattjj committed 6 years ago

Verified af064bb7

try a tweak on Literal for more cache hits

mattjj committed 6 years ago

310103f5

allow more types to be jaxpr literals, fixes #772

mattjj committed 6 years ago

9c931dde

remove accidentally-checked-in file

mattjj committed 6 years ago

3c1b832e

remove accidentally-checked-in file

mattjj committed 6 years ago

7751ba44

loosen tanh grad test tolerance (2bf9d5a fix)

mattjj committed 6 years ago

9fab325a

fix error in previous commit

mattjj committed 6 years ago

d70985d3

move stax randn init convert_element_type later

mattjj committed 6 years ago

3b290457

further improve grad-of-nonscalar error message

mattjj committed 6 years ago

5a049fee

loosen tolerance on tanh grad test

mattjj committed 6 years ago

2bf9d5aa

bring save, savez, load into jax.numpy namespace

mattjj committed 6 years ago

8eb10835

improve error message for grad of non-scalar funs

mattjj committed 6 years ago

74372732

Merge pull request #778 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 860cd8d5

Newer Older