ROCm/jax

Pull Requests Commits

add tuple simplification logic

mattjj committed 6 years ago

96775a4d

Merge pull request #865 from google/fix-reshape-grad-bug

mattjj committed 6 years ago

Verified a1219f10

Merge pull request #864 from jlebar/genrule-warning

mattjj committed 6 years ago

Verified 54ac9ee9

fix reshape transpose bug (and add tests)

mattjj committed 6 years ago

fef68dee

Fix bazel warning about jax_to_hlo genrule.

Justin Lebar committed 6 years ago

0cf33e96

Merge pull request #860 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 43d987e3

Use lax.full to create zeros array in gather transpose rule.

hawkinsp committed 6 years ago

129e73a2

Merge pull request #857 from jlebar/genrule-improvements

mattjj committed 6 years ago

Verified fbe9affb

Improvements to jax_to_hlo genrule.

Justin Lebar committed 6 years ago

0a770d97

Merge pull request #855 from google/scan-debug

mattjj committed 6 years ago

Verified 1dc4a4d0

Merge pull request #853 from jlebar/hlo-genrule

mattjj committed 6 years ago

Verified dd9186ae

fix the david sussillo bug

mattjj committed 6 years ago

191206db

Add jax_to_hlo.

Justin Lebar committed 6 years ago

42d0fc5d

Merge pull request #843 from mirca/patch-1

mattjj committed 6 years ago

Verified a6192644

Merge pull request #846 from jekbradbury/dot_batch_rule

mattjj committed 6 years ago

Verified 783893fd

Fix dot batch rule bug and add test + check

jekbradbury committed 6 years ago

edfe5203

Bump jaxlib version to 0.1.20 and update WORKSPACE

skye committed 6 years ago

619b0334

doc: partial class not needed in code snippet in README.md

mirca committed 6 years ago

Verified 5232428e

Merge pull request #840 from google/random-no-op-by-op

mattjj committed 6 years ago

Verified f237a242

Merge pull request #841 from jlebar/common-gotchas-fastmath

mattjj committed 6 years ago

Verified 63543787

Remove mention of CPU fastmath from common-gotchas notebook.

Justin Lebar committed 6 years ago

9794afeb

Update WORKSPACE

skye committed 6 years ago

3ef7693e

Bump jaxlib version and WORKSPACE

skye committed 6 years ago

075db4a7

make threefry_2x32 not do any op-by-op stuff

mattjj committed 6 years ago

a56a7d02

bump version for pypi

mattjj committed 6 years ago

9211bf39

fix build error

mattjj committed 6 years ago

0683e7e0

fix jax.rst docs (remove defvjp2 / defjvp2)

mattjj committed 6 years ago

bbf625e0

Merge pull request #818 from google/custom-transforms

mattjj committed 6 years ago

Verified 581d24c6

implement reviewer suggestions

mattjj committed 6 years ago

7c308dca

fix jax.numpy.transpose arg name 'axes'

mattjj committed 6 years ago

aebf7eb0

Older