ROCm/jax

Pull Requests Commits

Add defensive tuple() in lax.reduce_window

jekbradbury committed 5 years ago

Verified 23e01922

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

lexicographic sort_p: accept num_keys rather than comparator (#3715)

jakevdp committed 5 years ago

Verified 60d85277

Remove some non-inclusive language (#3710)

jakevdp committed 5 years ago

Verified d2f9c46a

fix flake error (#3716)

jakevdp committed 5 years ago

Verified 24c9ee64

Turn off INFO logging (again). (#3707)

skye committed 5 years ago

Verified 44eae610

Add jit to jax.image.resize (#3714)

hawkinsp committed 5 years ago

Verified 417de0d3

Add jax.image.resize. (#3703)

hawkinsp committed 5 years ago

Verified b943b31b

Generalize lax.sort to support lexicographic sorts. (#3709)

jakevdp committed 5 years ago

Verified 804e4493

Add _NOT_IMPLEMENTED attribute to jax.numpy (fixes #3689) (#3698)

jakevdp committed 5 years ago

Verified 0a6b715c

Add simple JAX API microbenchmarks. (#3679)

chr1sj0nes committed 5 years ago

Verified c1aeb8b3

Cleanup: record names in get_module_functions (#3697)

jakevdp committed 5 years ago

Verified b813ae3a

Cleanup: use test_util dtypes where possible (#3695)

jakevdp committed 5 years ago

Verified 19adce59

Fix a link to TensorBoard's profiler in Profiling JAX Programs (#3692)

8bitmp3 committed 5 years ago

Verified 11b40fb4

Added concurrent id_tap tests, disabled for GPU (#3690)

gnecula committed 5 years ago

Verified fdd7f0c8

Cleanup: define type lists in test_util & use in several test files. (#3616)

jakevdp committed 5 years ago

Verified 6b471e2a

Revert #3610 & #3684 (#3688)

jakevdp committed 5 years ago

Verified 82dbaca0

fix bad pmap tests from #3675 (#3685)

mattjj committed 5 years ago

Verified 1034f29d

fix pmap test on GPU/TPU (#3682)

jakevdp committed 5 years ago

Verified e89860c9

Older