ROCm/jax

Pull Requests Commits

stop_gradient_p -> ad_util.py, re-enable some mypy

mattjj committed 5 years ago

70c65231

disable mypy checks causing new errors

mattjj committed 5 years ago

903010b7

in custom_jvp/vjp stop_gradient on nondiff_argnums (#2804)

mattjj committed 5 years ago

Verified 8ccb907d

skip pinv test on tpu because no svd

mattjj committed 5 years ago

6b5e3676

use _max instead of max, fix #2795 (#2803)

mattjj committed 5 years ago

Verified 4eed4969

add tanh rule (#2653)

jacobjinkelly committed 5 years ago

Verified 59bdb1fb

Add explicit derivative for jax.numpy.linalg.pinv. (#2794)

hawkinsp committed 5 years ago

Verified 8fe3c59c

Merge pull request #2801 from google/scipy-conv-test-tolerance

jakevdp committed 5 years ago

Verified c3ab1fc5

loosen scipy convolve test tolerance (GPU flaky)

mattjj committed 5 years ago

fcb6ff0a

small typo fix (#2799)

yufengg committed 5 years ago

Verified aee18c88

Switch jaxlib Python code to use the lower-level xla.ops API when building XLA ops. (#2798)

hawkinsp committed 5 years ago

Verified 3f8c7359

attempt to fix failing travis (numerical issues)

mattjj committed 5 years ago

7334f97d

Added clearer error message for tracers in numpy.split (#2508)

gnecula committed 5 years ago

Verified a2c06d61

update travis to match min jaxlib version

mattjj committed 5 years ago

2e34dbc1

bump min jaxlib version (thanks @hawkinsp)

mattjj committed 5 years ago

6239e594

adjust test tolerance for tpu

mattjj committed 5 years ago

f753845f

changelog fixes

mattjj committed 5 years ago

db5d6193

factor out process_map / post_process_map (#2788)

mattjj committed 5 years ago

Verified b1cb3a5d

apply is_stable=True to sort translation rules (#2789)

mattjj committed 5 years ago

Verified 1bcaef14

Merge pull request #2786 from google/einsum-optimize-default

mattjj committed 5 years ago

Verified ec03f8e2

Bump jaxlib version in README to 0.1.45

skye committed 5 years ago

6db1f0c0

try optimize=True with einsum

mattjj committed 5 years ago

783e24fd

attempt to fix changelog formatting bugs

mattjj committed 5 years ago

18f96742

autodiff cookbook: assume continuous second derivatives

mattjj committed 5 years ago

Verified 964cf4fb

Bump jaxlib version to 0.1.45 and update WORKSPACE and CHANGELOG. (#2785)

skye committed 5 years ago

Verified 4703b0fa

Merge pull request #2752 from apaszke/ad-fixme

mattjj committed 5 years ago

Verified 0d7d6637

Merge branch 'master' into ad-fixme

mattjj committed 5 years ago

Verified 18b23404

Merge pull request #2763 from jacobjinkelly/adamax

mattjj committed 5 years ago

Verified f527ed41

update version and changelog for pypi

mattjj committed 5 years ago

4433ddfb

also skip jax.numpy.unique test on tpu

mattjj committed 5 years ago

94178d98

Older