ROCm/jax

Pull Requests Commits

fix bug in categorical test, disable #3611 on tpu (#3633)

mattjj committed 5 years ago

Verified 65c4d755

Enable XLA SPMD partitioning by default. (#3151)

skye committed 5 years ago

Verified 0f7b6444

skip gamma tests on tpu for compilation speed (#3631)

mattjj committed 5 years ago

Verified 1bd04e2d

Install scikit-image and add try;except around from google.colab import files import statement as google.colab is not available locally and in kokoro (#3626)

yashk2810 committed 5 years ago

Verified aaf0c4e3

skip some ode tests on gpu for speed (#3629)

mattjj committed 5 years ago

Verified ba1b5ce8

Add try; except to cells that fail and print out the error message so that the notebook can execute end-to-end (#3624)

yashk2810 committed 5 years ago

Verified 236a5881

Update XLA. (#3623)

hawkinsp committed 5 years ago

Verified 47ac6122

Reimplement argmin/argmax using a single pass variadic reduction. (#3611)

hawkinsp committed 5 years ago

Verified 141fabbb

improve vmap axis spec structure mismatch errors (#3619)

mattjj committed 5 years ago

Verified 107689e9

fix reduction repeated axis error (#3618)

mattjj committed 5 years ago

Verified eb2a2275

add jit-consistency checks to a few lax_numpy tests

froystig committed 5 years ago

e808681f

Rework type support for lax cumulative reductions (#3609)

jakevdp committed 5 years ago

Verified db8f66d5

Fix shape rule for lax.pad for input dimensions of size 0. (#3608)

hawkinsp committed 5 years ago

Verified 420ef4e0

[jax2tf] Re-enable JIT (#3605)

gnecula committed 5 years ago

Verified 7ecb441d

[jax2tf] Fix tf.gather handling of out-of-bounds indices (#3604)

gnecula committed 5 years ago

Verified 59344028

Retain original docstring when vmap'ing functions. (#3592)

tomhennigan committed 5 years ago

Verified 2ef39abf

Cleanup: remove some test interdependence (#3600)

jakevdp committed 5 years ago

Verified 09d128ed

Use precision=HIGHEST in expm repeated squaring (#3601)

jekbradbury committed 5 years ago

Verified 039b6e2e

Merge pull request #3588 from ekelsen/master

jakevdp committed 5 years ago

Verified ec21a3f5

Add ability to specify individual test targets via a regex (#3549)

jakevdp committed 5 years ago

Verified a9fad49e

add source info to jaxpr typechecking messages

froystig committed 5 years ago

101581d8

bool -> bool_ for reasons that make no sense, (bool used to be any?!)

Erich Elsen committed 5 years ago

aa6585f9

add support bool identity values

Erich Elsen committed 5 years ago

b46bd230

Use re.search and include test class name

jakevdp committed 5 years ago

e9aac7bb

fix missing imports

jakevdp committed 5 years ago

c62bad5f

Add ability to specify individual test targets

jakevdp committed 5 years ago

afce718e

Merge branch 'master' of github.com:ekelsen/jax

Erich Elsen committed 5 years ago

acc95028

change ending tick mark style

Erich Elsen committed 5 years ago

77a023df

Merge branch 'master' into master

ekelsen committed 5 years ago

Verified b362fd04

[jax2tf] Add support for custom JVP/VJP (#3581)

gnecula committed 5 years ago

Verified 6147026b

Newer Older