ROCm/jax

Pull Requests Commits

revive old numpy promotion logic

mattjj committed 6 years ago

abe4f613

Update Jaxlib references to 0.1.16.

hawkinsp committed 6 years ago

d63b8a47

fix stax dtype breakage

mattjj committed 6 years ago

913bc6b5

mattjj committed 6 years ago

1a908b0f

fix stax initialization rng bug, remove temp file

mattjj committed 6 years ago

8ffb9417

tweak docstrings in mnist examples

mattjj committed 6 years ago

92e5f93a

bump version for pypi

mattjj committed 6 years ago

5d902298

improve scan error message when output is not pair

mattjj committed 6 years ago

c3066794

Merge pull request #761 from google/issue756

mattjj committed 6 years ago

Verified 2409a74f

address reviewer comments

mattjj committed 6 years ago

c743c14c

Merge pull request #760 from google/issue758

mattjj committed 6 years ago

Verified 0490d06a

make jax.random default dtypes 64-bit

mattjj committed 6 years ago

93e201f8

make a batching.py utility function more defensive

mattjj committed 6 years ago

86ad8e04

patch utility functions in batching.py

mattjj committed 6 years ago

09b22996

when xla_computation sees no kwargs, don't make ()

mattjj committed 6 years ago

a193b359

Merge pull request #759 from google/issue757

mattjj committed 6 years ago

Verified cff08e50

compare binary lattice prefixes in scan

mattjj committed 6 years ago

0307ce58

Merge pull request #755 from google/scan-vmap

mattjj committed 6 years ago

Verified 3815662a

add vmap-of-scan tests, fix minor bugs

mattjj committed 6 years ago

4fe2bcfe

move control flow batching tests

mattjj committed 6 years ago

7065cb63

scan of vmap sketched out! cf. #716

mattjj committed 6 years ago

140699ed

Merge pull request #754 from google/issue746

mattjj committed 6 years ago

Verified ff481472

support both .reshape(*shape) and .reshape(shape)

mattjj committed 6 years ago

14d16b2d

Merge pull request #752 from google/issue751

mattjj committed 6 years ago

Verified 83a04c02

Merge pull request #750 from kho/master

mattjj committed 6 years ago

Verified 522cf0ae

fix negative index handling in index_take

mattjj committed 6 years ago

ec6e39b9

bump version for pypi

mattjj committed 6 years ago

42ea9fef

fix typo in lax.scan autodiff test (cf. #749)

mattjj committed 6 years ago

e04f344c

Merge pull request #749 from google/scan-debug

mattjj committed 6 years ago

Verified b706879d

Fix vmap() when there are empty tuples in the input

kho committed 6 years ago

8ea5eff3

Older