ROCm/jax

Pull Requests Commits

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

[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

[jax2tf] Relax tolerance for JaxPrimitiveTest.test_unary_elementwise_lgamma_float32 (#3767)

gnecula committed 5 years ago

Verified 23c279f0

[jax2tf] Fix error in sorting with TF graphs. (#3764)

bchetioui committed 5 years ago

Verified d55c47d2

[jax2tf] Refactor tests to increase coverage (#3700)

gnecula committed 5 years ago

Verified c380356f

Added installation note for jax2tf with GPU support. (#3750)

bchetioui committed 5 years ago

Verified 68c8dc78

Support differentiation through jax.lax.all_to_all (#3733)

jpuigcerver committed 5 years ago

Verified f02d5b46

Cleanup: convert uses of `import numpy as onp` in library code (#3754)

jakevdp committed 5 years ago

Verified a7c2cdea

Cleanup: convert uses of 'import numpy as onp' in tests (#3756)

jakevdp committed 5 years ago

Verified 512ed18d

Fix low probability (4/1000) flake of batching_test on GPU. (#3752)

hawkinsp committed 5 years ago

Verified 58aba9bc

A jit-able version of np.repeat. (#3670)

jg8610 committed 5 years ago

Verified f6f97554

Fix compilation bug in histogram_bin_edges (#3745)

jakevdp committed 5 years ago

Verified 06053212

lax_numpy: rename arguments to match numpy (#3747)

jakevdp committed 5 years ago

Verified e1f8f246

Make jnp.take work for empty slices of empty arrays. (#3751)

hawkinsp committed 5 years ago

Verified 7f9c2f1d

Add pshuffle to docs (#3742)

Chase Roberts committed 5 years ago

Verified 2b7a39f9

Fix tuple() in reduce_window padding (#3748)

jekbradbury committed 5 years ago

Verified f78ccf1c

Implement nearest neighbor image resizes. (#3743)

hawkinsp committed 5 years ago

Verified a017c108

Remove unused private function (#3744)

jakevdp committed 5 years ago

Verified 9a867ca6

Improve np.intersect1d (#3739)

aldragan0 committed 5 years ago

Verified a6ab742f

Older