ROCm/jax

Pull Requests Commits

froystig committed 5 years ago

4cdbbba7

Merge pull request #3782 from hawkinsp/dot

hawkinsp committed 5 years ago

Verified 9a01d78b

Merge pull request #3783 from hawkinsp/matmul

hawkinsp committed 5 years ago

Verified 3eda6e97

Merge pull request #3789 from hawkinsp/rw

hawkinsp committed 5 years ago

Verified 57c1822d

Merge pull request #3788 from hawkinsp/convtol

hawkinsp committed 5 years ago

Verified 7c4d41f0

Improve reduce-window testing.

hawkinsp committed 5 years ago

7e3433e2

Relax test tolerance for conv_general_dilated gradients.

hawkinsp committed 5 years ago

e04fdeb0

Merge pull request #3786 from hawkinsp/einsumargs

hawkinsp committed 5 years ago

Verified 7b3eff82

stack traces without jax-internal frames (#3674)

froystig committed 5 years ago

Verified f7260a47

Update numpy signatures test.

hawkinsp committed 5 years ago

252a027a

Avoid broadcasting for batch dimensions in jnp.matmul.

hawkinsp committed 5 years ago

8fc1332e

Use keyword arguments in einsum.

hawkinsp committed 5 years ago

8e986f9c

Don't move batch dimensions to start in jnp.einsum.

hawkinsp committed 5 years ago

6a37e26b

Relax dimension ordering rules for dot_general. (#3778)

hawkinsp committed 5 years ago

Verified 943c4216

Use iteration over equations to test for "transpose" and "broadcast".

hawkinsp committed 5 years ago

e4d5eade

Also test for transpose in dot vmap test.

hawkinsp committed 5 years ago

165f31ef

docstring for api_boundary

froystig committed 5 years ago

dd4db64e

disable false pytype error

froystig committed 5 years ago

733f6c4e

append filtered stack traces to error messages raised under transformations

froystig committed 5 years ago

6416ca0e

Relax dimension ordering rules for dot_general.

hawkinsp committed 5 years ago

e2e73a85

[x64 deprecation] Create _np_array utility routine (#3779)

jakevdp committed 5 years ago

Verified 3fb88742

Revert "Add `in_parts` and `out_parts` optional arguments `jax.xla_computation`. (#3771)" (#3780)

skye committed 5 years ago

Verified 44fbce56

Enable `all_to_all` in multi-host settings. (#3772)

skye committed 5 years ago

Verified bfe8e4f0

[jax2tf] Fix interface of ConvertAndCompare function. (#3776)

bchetioui committed 5 years ago

Verified 7b18a4e8

[jax2tf] Fix bug with double XLA compile (#3765)

gnecula committed 5 years ago

Verified 15125b87

Add `in_parts` and `out_parts` optional arguments `jax.xla_computation`. (#3771)

skye committed 5 years ago

Verified dbc3f83f

Change onp/np to np/jnp in docs & notebooks (#3760)

jakevdp committed 5 years ago

Verified 05904faf

Update scipy.ndimage.map_coordinates docstring (#3762)

claudiofantacci committed 5 years ago

Verified 150d028d

source sync (#3763)

jg8610 committed 5 years ago

Verified 34669d74

block-unrolled scan primitive implementation (#3738)

froystig committed 5 years ago

Verified 8a62a9b6

Older