ROCm/jax

Pull Requests Commits

Address review comments.

SiegeLordEx committed 6 years ago

3e0e2695

uff actually finish last change

levskaya committed 6 years ago

ee2c6cca

just say no to pop

levskaya committed 6 years ago

229fdfe9

Merge pull request #1245 from google/issue1236

mattjj committed 6 years ago

Verified 661aaf1d

add option to disable rank-promotion broadcasting

mattjj committed 6 years ago

d700716e

refactor backend check in central dispatch logic

levskaya committed 6 years ago

bd35ef23

Merge pull request #1243 from google/while-cond-const

mattjj committed 6 years ago

Verified af8859b9

levskaya committed 6 years ago

e60629fd

Clarify the intended purpose of tree_util.

SiegeLordEx committed 6 years ago

b1604459

fix last mistake

levskaya committed 6 years ago

ba572231

fix an overzealous error message w/ const cond_fun

mattjj committed 6 years ago

050ce59a

Merge pull request #1242 from google/nones-are-pytrees

mattjj committed 6 years ago

Verified 36539f1f

Update build_mac_jaxlib.yml

skye committed 6 years ago

Verified 003eda84

be careful with params.pop

levskaya committed 6 years ago

dbb68bd1

restore the behavior that Nones are pytrees

mattjj committed 6 years ago

e66582e8

Update build_mac_jaxlib.yml

skye committed 6 years ago

Verified 3b303a50

Update build_mac_jaxlib.yml

skye committed 6 years ago

Verified 24efaa77

Remove version restriction from opt_einsum.

skye committed 6 years ago

90093b78

Merge pull request #1238 from google/jb/cond

jekbradbury committed 6 years ago

Verified aab71eae

levskaya committed 6 years ago

4811231e

batching rule for cond

jekbradbury committed 6 years ago

7830cede

Testing out github workflows

skye committed 6 years ago

Verified e025fac1

Merge pull request #1240 from google/issue1239

mattjj committed 6 years ago

Verified 19bb5639

add flag for logging when jit performs compilation

mattjj committed 6 years ago

98c7567a

ensure DeviceArray.aval is ShapedArray

mattjj committed 6 years ago

7539325f

Merge pull request #1237 from fehiepsi/fix-cond

mattjj committed 6 years ago

Verified 960a56ae

try not to sabotage all jax users, use normalized platform names to control device_put behavior.

levskaya committed 6 years ago

ceca1b7a

Merge branch 'master' into multibackend

levskaya committed 6 years ago

8a78c92b

fix default kwarg issue

levskaya committed 6 years ago

78c04fe3

fix kwarg order bug for device constant put handler

levskaya committed 6 years ago

8bbe1b0c

Newer Older