ROCm/jax

Pull Requests Commits

add missing space in error message

mattjj committed 5 years ago

57d3fd87

Merge pull request #4732 from shoyer:rk45-ref

a-googler committed 5 years ago

09d87efb

Add reference on RK45

shoyer committed 5 years ago

3c5f82f3

Merge pull request #4706 from apaszke:vmap-collectives-in-scan

a-googler committed 5 years ago

d158647c

Merge pull request #4691 from bchetioui:fix_select_and_scatter_add

a-googler committed 5 years ago

278d4131

[jax2tf] Fix implementation and tests of select_and_scatter_add conversion.

bchetioui committed 5 years ago

dfce7482

Merge pull request #4731 from jakevdp:weak-promotion-test

a-googler committed 5 years ago

3e6da7e0

Add explicit tests of weak/strong promotion semantics

jakevdp committed 5 years ago

7cd319a0

Merge pull request #4727 from gnecula:tf_poly5

a-googler committed 5 years ago

77f3c225

Enable C++ jit again.

jblespiau committed 5 years ago

dad8377b

[jax2tf] Expanding jax2tf shape polymorphism.

gnecula committed 5 years ago

137ec99c

Merge pull request #4720 from google:static-argnums-immutable

a-googler committed 5 years ago

98fb6cfc

Merge pull request #4726 from google:remove-unused-import

a-googler committed 5 years ago

a2b91fad

remove unused import

mattjj committed 5 years ago

1c4d873f

Raise an error on non-hashable static arguments for jax.jit and xla_computation.

jblespiau committed 5 years ago

cb48f423

tiny change to test source sync

froystig committed 5 years ago

a7de6942

Merge pull request #4597 from gnecula:tf_poly3

froystig committed 5 years ago

3e56be2a

mention static_argnum values should be immutable

mattjj committed 5 years ago

1da36943

Merge pull request #4716 from gnecula:update_changelog

a-googler committed 5 years ago

38ba3471

Update CHANGELOG

gnecula committed 5 years ago

da9a64ef

gnecula committed 5 years ago

8458cab0

Edits for the documentation

gnecula committed 5 years ago

36542e04

[jax2tf] Expanding jax2tf shape polymorphism.

gnecula committed 5 years ago

f0d2a4d2

Merge pull request #4713 from google:simpler-select-jvp

a-googler committed 5 years ago

9092ad3d

simplify select jvp

mattjj committed 5 years ago

7a73e99e

Merge pull request #4708 from google:docfix-jit

a-googler committed 5 years ago

ab3a4c2f

Merge pull request #4695 from Joshuaalbert:master

a-googler committed 5 years ago

a4956695

all changes plus test verifcation on TPU squashed

Joshuaalbert committed 5 years ago

231168d4

Merge pull request #4707 from google:traceback-register

a-googler committed 5 years ago

25b40702

Merge pull request #4709 from google:update-pypi

a-googler committed 5 years ago

4c9545d6

Older