ROCm/jax

Pull Requests Commits

linalg_test: define test matrices lazily (#3684)

jakevdp committed 5 years ago

Verified 2be1baa4

AssertionError -> ValueError

jakevdp committed 5 years ago

9c3e6c30

fix pmap test on GPU/TPU

jakevdp committed 5 years ago

4711589c

Remove a deprecated reference to testExamplesJaxprDoc in Understanding Jaxpr (#3680)

8bitmp3 committed 5 years ago

Verified 242b382b

Make infeed_test and host_callback_test independent. (#3676)

gnecula committed 5 years ago

Verified bf97e479

fix ppermute test bugs found by @jekbradbury (#3675)

mattjj committed 5 years ago

Verified d2ebb6eb

support multiple_results and custom JVPs in jet (#3657)

mattjj committed 5 years ago

Verified e5ba5f1e

Rename swish to silu, add swish as an alias to silu. (#3673)

hawkinsp committed 5 years ago

Verified a2fb798d

Make LU gradient work for low-rank matrices (#3610)

dpfau committed 5 years ago

Verified 23deefa7

Update XLA. (#3671)

hawkinsp committed 5 years ago

Verified c6382644

Refine argument validation inside jax.scipy.sparse.linalg.cg (#3630)

shoyer committed 5 years ago

Verified 36eb137d

Cleanup last use of msgpack. (#3668)

gnecula committed 5 years ago

Verified 2a9c2d22

Improve jax_debug_nans documentation (#3665)

IgorWilbert committed 5 years ago

Verified c485a5b0

Fix broken links to deleted notebook (#3663)

gnecula committed 5 years ago

Verified 8f936073

Add an empty line to JAX change log (#3661)

8bitmp3 committed 5 years ago

Verified 3bbe06e8

Refactored host_callback to use the C++ runtime. (#3644)

gnecula committed 5 years ago

Verified 4f3011f3

Correctly handle bfloat16 input and output from jax2tf functions. (#3652)

tomhennigan committed 5 years ago

Verified 904f34a9

Adds information about iterator in fori_loop in The Sharp Bits documentation (#3632)

IgorWilbert committed 5 years ago

Verified 3d59a0bc

[jax2tf] Fix out-of-bounds errors for dynamic slices (#3649)

gnecula committed 5 years ago

Verified 1e7e5c57

improve concreteness error message for nn.one_hot (#3656)

mattjj committed 5 years ago

Verified 49cfe268

make psum transpose handle zero cotangents (#3653)

mattjj committed 5 years ago

Verified 796df9c5

[jax2tf] Add support for first-order AD to converted functions (#3593)

gnecula committed 5 years ago

Verified 7f7fd997

Add support for 0d transpose convolution (#3643)

romanngg committed 5 years ago

Verified 4442c333

fix prng key reuse in differential privacy example (#3646)

mattjj committed 5 years ago

Verified d10cf0e3

Updated minimum jaxlib to 0.1.51 (#3642)

gnecula committed 5 years ago

Verified 166e795d

Change the version of tensorflow, the one in jaxlib 0.1.51 does not build (#3637)

gnecula committed 5 years ago

Verified bfe8f873

Release jaxlib 0.1.51 (#3636)

gnecula committed 5 years ago

Verified a5ed1615

[jax2tf] Update the tf-nightly version to 2020-07-01 (#3635)

gnecula committed 5 years ago

Verified 448c635a

Add test of jax.numpy vs numpy call signatures (#3628)

jakevdp committed 5 years ago

Verified 65c1328d

Add CI tests for numpy 1.16.4 (#3634)

jakevdp committed 5 years ago

Verified f1ec4beb

Newer Older