ROCm/jax

Pull Requests Commits

Merge branch 'master' into linear-transpose

shoyer committed 5 years ago

b20f0cd9

shoyer committed 5 years ago

9f09e65d

implement jnp.apply_over_axes (#4225)

jakevdp committed 5 years ago

Verified e18a9731

[jax2tf] More convolution test disabling (#4304)

gnecula committed 5 years ago

Verified dcaa28c6

[jax2tf] Disable some convolution tests (#4303)

gnecula committed 5 years ago

Verified a433c16f

[jax2tf] Add tests for the conversion of conv_general_dilated (#4222)

bchetioui committed 5 years ago

Verified 1f95414f

[jax2tf] Enable testing for SVD on TPU for float16 (#4288)

gnecula committed 5 years ago

Verified a9430561

Fix code rendering in optimizers documentation (#4296)

ed1d1a8d committed 5 years ago

Verified 1aeaa4a0

Mention in docstring when function is not implemented (#4297)

jakevdp committed 5 years ago

Verified 19201f4b

Fix device_put_sharded() for concrete values (#4298)

jakevdp committed 5 years ago

Verified 2811b9b2

Add api.device_put_sharded() (#4287)

jakevdp committed 5 years ago

Verified 95287a05

tweak traceback for unbound axis names (#4295)

mattjj committed 5 years ago

Verified 5520948b

jnp.divide: remove obsolete condition for py2 behavior (#4286)

jakevdp committed 5 years ago

Verified 9c393812

omnistaging on by default (#4038)

mattjj committed 5 years ago

Verified 2678a464

update version and changelog for pypi (#4294)

mattjj committed 5 years ago

Verified 6af47690

Lower LU decomposition to a custom TPU implementation for float32 types. (#4291)

hawkinsp committed 5 years ago

Verified cefa93f2

Modifies eig_p and related operations to take advantage of the new jaxlib geev API (#4266)

bchetioui committed 5 years ago

Verified 58a117fe

[jax2tf] Added support for x64 for the remaining test files (#4282)

bchetioui committed 5 years ago

Verified a5c2c472

[jax2tf] Build a primitive harness for test_type_promotion. (#4279)

bchetioui committed 5 years ago

Verified 4e04d4ea

Allow device_get to pass Python scalars through unchanged (#4283)

trevorcai committed 5 years ago

Verified 90403367

revert #4277 (google failure) (#4281)

mattjj committed 5 years ago

Verified 7569e800

Simplify the interface for host_callback.id_tap (#4101)

shoyer committed 5 years ago

Verified 6bd3216b

[jax2tf] Fix tests when running with JAX_ENABLE_X64=1. (#4261)

bchetioui committed 5 years ago

Verified 2ff34792

[jax2tf] Add converted primitives without tests to the generated doc. (#4248)

bchetioui committed 5 years ago

Verified 7e6d114f

[jax2tf] Added the last comments from the jax2tf doc inside the (#4249)

bchetioui committed 5 years ago

Verified fa827a59

[jax2tf] Fix lax.div_p (#4263)

petebu committed 5 years ago

Verified 6e2fa39e

Avoid rank promotion in stax biases (#4277)

romanngg committed 5 years ago

Verified 9fc4353c

Avoid rank promotion in np.outer (#4276)

romanngg committed 5 years ago

Verified 38b43ef9

fixing typo (#4273)

alexminnaar committed 5 years ago

Verified 64bead20

thread backend in pxla.replicate (#4272)

mattjj committed 5 years ago

Verified f039f6da

Older