ROCm/jax

Pull Requests Commits

add option to disable rank-promotion broadcasting

mattjj committed 6 years ago

d700716e

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

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

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

fix cond with true_consts

fehiepsi committed 6 years ago

9f876c48

Merge pull request #1235 from google/vmap-axes-lists

mattjj committed 6 years ago

Verified 985ecbe5

allow vmap in_axes to be lists

mattjj committed 6 years ago

de6ce2a5

bump version for pypi

mattjj committed 6 years ago

0c832b2d

Merge pull request #1231 from hgt312/fix_readme_type

mattjj committed 6 years ago

Verified 1919ba51

hgt312 committed 6 years ago

Verified 09a2d068

Merge pull request #1229 from google/no-more-tuples-fixes

mattjj committed 6 years ago

Verified d7894198

minor fixes from trax, revise eval_shape api

mattjj committed 6 years ago

85179975

Merge pull request #1217 from shoyer/doc-index-page

mattjj committed 6 years ago

Verified 1e375daf

fix ShardedDeviceArray.__getitem__

mattjj committed 6 years ago

7323e154

update tree_util_tests w/ None not a pytree

mattjj committed 6 years ago

06b5c402

Merge pull request #1227 from google/sigmoid-expit

mattjj committed 6 years ago

Verified 9f46fc4b

make stax.sigmoid call scipy.special.expit

mattjj committed 6 years ago

765af977

Merge pull request #1224 from google/no-more-tuples

mattjj committed 6 years ago

Verified a8e0c255

remove pat_fmap

mattjj committed 6 years ago

f56312bb

De-tuplify the rest of the core

mattjj committed 6 years ago

b702f8de

Older