ROCm/jax

Pull Requests Commits

Avoid dynamic slicing in threefry implementation.

hawkinsp committed 6 years ago

719e17ba

Merge pull request #1186 from hawkinsp/master

hawkinsp committed 6 years ago

Verified a36c0829

Remove unnecessary reshape/concatenate in dynamic_slice_in_dim.

hawkinsp committed 6 years ago

932877dd

Fix Python 2 compatibility.

hawkinsp committed 6 years ago

099354aa

Merge pull request #1187 from brianwa84/patch-2

hawkinsp committed 6 years ago

Verified bd441998

Address review comment.

hawkinsp committed 6 years ago

e28e73b3

Ensure reps is a tuple (allows list or other iterable)

brianwa84 committed 6 years ago

Verified 4b693777

Fix batching rule.

hawkinsp committed 6 years ago

e57a5c42

Fix batching rule.

hawkinsp committed 6 years ago

e4a7d307

Change dynamic-slice and dynamic-update-slice primitives to have one argument per index, not a single array index.

hawkinsp committed 6 years ago

d09924f7

Merge pull request #1184 from majnemer/remainder

hawkinsp committed 6 years ago

Verified 7db288eb

Use lax.rem less often in remainder

majnemer committed 6 years ago

079ded40

remove author info in notebook (old, redundant)

mattjj committed 6 years ago

Verified 2dec779b

Disable correct TPU test.

hawkinsp committed 6 years ago

4559d36d

Merge pull request #1183 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 4671db63

Enable some tests that now seem to pass.

hawkinsp committed 6 years ago

d2f2da29

Merge pull request #1178 from hawkinsp/docker

hawkinsp committed 6 years ago

Verified 65034853

Merge pull request #1182 from pifon2a/master

hawkinsp committed 6 years ago

Verified 56c2008b

pifon2a committed 6 years ago

92a33b8b

Merge pull request #1180 from brianwa84/patch-1

froystig committed 6 years ago

Verified bb3882a3

Make DeviceValue and subclasses weakref friendly

brianwa84 committed 6 years ago

Verified 8718e305

Update Docker build to produce manylinux2010 compliant wheels for non-cuda builds.

hawkinsp committed 6 years ago

61713fe5

Merge pull request #1175 from google/issue1172

mattjj committed 6 years ago

Verified eb2ddb4b

Readd coding declaration to api.py

hawkinsp committed 6 years ago

c84cb34a

Remove spurious coding line from api.py

hawkinsp committed 6 years ago

f898ac1a

address reviewer comments, no op-by-op in threefry

mattjj committed 6 years ago

75bb38e7

Merge pull request #1174 from hawkinsp/master

hawkinsp committed 6 years ago

Verified c62baf92

Merge pull request #1149 from shoyer/custom-implicit-solve

mattjj committed 6 years ago

Verified f2fe49f9

improve prng compile times with outer-loop rolling

mattjj committed 6 years ago

d857d3f5

Merge pull request #1176 from hawkinsp/argsplit

hawkinsp committed 6 years ago

Verified 3db85b0b

Newer Older