ROCm/jax

Pull Requests Commits

[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

Add defensive tuple() in lax.reduce_window (#3741)

jekbradbury committed 5 years ago

Verified 6017205c

Implement complex convolutions on CPU and GPU. (#3735)

hawkinsp committed 5 years ago

Verified 3c6cd5fb

[jax2tf] First draft of converting sort_p using TF2XLA (#3713)

bchetioui committed 5 years ago

Verified 6391cfe7

Generalize reduce-window padding to support (lo, hi) pairs. (#3728)

hawkinsp committed 5 years ago

Verified 71253ac4

Fix shape error when taking JVP of reduce-prod over size 0 axis. (#3729)

hawkinsp committed 5 years ago

Verified a9da06ce

Implement np.intersect1d (#3726)

aldragan0 committed 5 years ago

Verified 0d81e988

Change image resize implementation to use a matmul per dimension. (#3720)

hawkinsp committed 5 years ago

Verified 9da9156b

check matmul inputs aren't scalar (#3725)

mattjj committed 5 years ago

Verified 51ca57d5

Import profiler in jax/__init__.py (#3719)

tomhennigan committed 5 years ago

Verified aa75209d

Merge pull request #3705 from NeilGirdhar/nestable_vjp

shoyer committed 5 years ago

Verified eb67571b

Make vjp cotangent functions pytree-like

NeilGirdhar committed 5 years ago

503e5973

Add a jax.profiler.StepTraceContext API. (#3591)

Jiho Choi committed 5 years ago

Verified e073e254

hfft and ihfft implementation (#3664)

aldragan0 committed 5 years ago

Verified 412b9d52

Newer Older