ROCm/jax

Pull Requests Commits

add initial jet tests

mattjj committed 6 years ago

3e29c9fb

intial jet xform (higher-order taylor series ad)

mattjj committed 6 years ago

060c8edd

Merge branch 'master' of github.com:google/jax

mattjj committed 6 years ago

fd4b84bd

fix np.var dtype bug

mattjj committed 6 years ago

3168006f

Disable gradient test for entr.

hawkinsp committed 6 years ago

f8fa1a60

More build formatting fixes.

hawkinsp committed 6 years ago

6bc47626

Build formatting fixes.

hawkinsp committed 6 years ago

e0b31ac3

Merge pull request #1099 from hawkinsp/cusolver

hawkinsp committed 6 years ago

Verified e2d6d3ac

Remove commented lines from test.

hawkinsp committed 6 years ago

99735958

Add support for linear algebra ops on GPU using Cusolver:

hawkinsp committed 6 years ago

ed3e2308

Merge pull request #1098 from hawkinsp/fixes

hawkinsp committed 6 years ago

Verified 7c060435

Fix test breakage at head.

hawkinsp committed 6 years ago

b45d1ec1

Merge pull request #1094 from fehiepsi/mvngamma

mattjj committed 6 years ago

Verified 1dfdd8da

Merge pull request #1091 from fehiepsi/tril

mattjj committed 6 years ago

Verified 1f3b4ae9

Merge pull request #1089 from fehiepsi/nonlinear

mattjj committed 6 years ago

Verified 76732852

Merge pull request #1088 from fehiepsi/median

mattjj committed 6 years ago

Verified fd98f957

Merge pull request #1086 from fehiepsi/sort

mattjj committed 6 years ago

Verified e597b222

Merge pull request #1083 from j-towns/patch-1

mattjj committed 6 years ago

Verified db8116d0

Merge pull request #1087 from ibab/master

mattjj committed 6 years ago

Verified 0134e2e2

Merge pull request #1096 from hawkinsp/fixes

hawkinsp committed 6 years ago

Verified 1f267cf6

Add --spawn_strategy from TensorFlow configuration.

hawkinsp committed 6 years ago

40c517e9

Merge pull request #1095 from hawkinsp/fixes

hawkinsp committed 6 years ago

Verified 24d4aaf3

Fix test failures due to Numpy 1.17.

hawkinsp committed 6 years ago

021d93a3

add multigammaln, entr

fehiepsi committed 6 years ago

7f4dc87a

expose tril_indices triu_indices

fehiepsi committed 6 years ago

7a5aecea

Merge pull request #1090 from hawkinsp/pytree

hawkinsp committed 6 years ago

Verified dce27bfc

Fix test failures.

hawkinsp committed 6 years ago

c302e388

expose sigmoid too

fehiepsi committed 6 years ago

5ffe2ae5

Merge pull request #1085 from hawkinsp/pytree

hawkinsp committed 6 years ago

Verified 129af6a5

Address review comments.

hawkinsp committed 6 years ago

cb53ca87

Older