ROCm/jax

Pull Requests Commits

add more ode tests

mattjj committed 5 years ago

91eaa24b

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

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

Older