ROCm/jax

Pull Requests Commits

add test for #2822

mattjj committed 5 years ago

d7e2206b

Merge branch 'master' into mapped-invars-fixes

mattjj committed 5 years ago

c77582ce

only maximally stage out for some call primitives (#2834)

mattjj committed 5 years ago

Verified 8f902452

skip broken test

mattjj committed 5 years ago

a649fc0c

Fix lax.rng_uniform. (#2830)

malmaud committed 5 years ago

Verified 77901e9f

Remove platform canonicalization from xla_bridge.py (#2815)

skye committed 5 years ago

Verified 343e486d

handle mapped_invars correctly in more places

mattjj committed 5 years ago

7c7a0ede

add more ode tests (#2819)

mattjj committed 5 years ago

Verified 11d7fb05

add ode test file (#2818)

mattjj committed 5 years ago

Verified 6ad2908f

Fix time issues in odeint reverse mode (#2817)

samuela committed 5 years ago

Verified 23f4874b

Feature/permutation (#1568)

MichaelMarien committed 5 years ago

Verified e0d42e90

implement jet rules by lowering to other primitives (#2816)

jacobjinkelly committed 5 years ago

Verified fc4203c3

use static_argnums in xla_computation (#2812)

mattjj committed 5 years ago

Verified 25183436

Pin mypy version in .travis.yml. (#2811)

skye committed 5 years ago

Verified 9b6976bf

Make ShardedDeviceArray._value threadsafe again. (#2810)

skye committed 5 years ago

Verified 46306493

Remove usage of xla_client.{Computation,ComputationBuilder}. (#2808)

hawkinsp committed 5 years ago

Verified 5290c03a

rewrite axis_index implementation, use custom bind (#2807)

mattjj committed 5 years ago

Verified d2653a1e

stop_gradient_p -> ad_util.py, re-enable some mypy (#2806)

mattjj committed 5 years ago

Verified 13a17286

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

Older