ROCm/jax

Pull Requests Commits

resolve merge conflicts with master

levskaya committed 6 years ago

685ca676

fix cond with true_consts

fehiepsi committed 6 years ago

9f876c48

Merge branch 'master' into multibackend

levskaya committed 6 years ago

10e0842f

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

Added basic behavior unit tests of multibackend jit.

levskaya committed 6 years ago

c839c6a6

minor fixes from trax, revise eval_shape api

mattjj committed 6 years ago

85179975

Error on nested conflicting explicit jit backend specifications.

levskaya committed 6 years ago

88a1fbf6

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

Update GPU memory documentation.

skye committed 6 years ago

c50918fe

Merge pull request #1225 from brianwa84/patch-6

mattjj committed 6 years ago

Verified 1d66650c

Cope with old numpy lacking axis arg

brianwa84 committed 6 years ago

Verified 451ff2d6

Update linearize to no-tuple version

dougalm committed 6 years ago

3c37a326

Some progress de-tupling ad.py

dougalm committed 6 years ago

c53c8bbb

De-dup equations with multiple lhs vars when creating a jaxpr

dougalm committed 6 years ago

20fad746

Start exploring jaxprs without tuples

dougalm committed 6 years ago

6d71396d

Merge pull request #1221 from j-towns/jaxpr-hash

mattjj committed 6 years ago

Verified 66c6b899

Implement Jaxpr __hash__

j-towns committed 6 years ago

9b19afd4

add multibackend option to soft_pmap

levskaya committed 6 years ago

dbe4bdf9

Newer Older