ROCm/jax

Pull Requests Commits

Make op-by-op work with all jit-returned devicearrays.

levskaya committed 6 years ago

f01fc35c

Fix index of API docs

shoyer committed 6 years ago

ee54f655

fix missing pop default

levskaya committed 6 years ago

75298156

levskaya committed 6 years ago

f7c1517d

small fix for multibackend

levskaya committed 6 years ago

29c6fe88

WIP: experimental multibackend jit

levskaya committed 6 years ago

cc87fb60

Require opt_einsum version to be less than 3.0.0.

skye committed 6 years ago

921096e3

Update jaxlib version and XLA.

skye committed 6 years ago

47207760

Merge pull request #1207 from majnemer/elu

mattjj committed 6 years ago

Verified 4e52c432

Use expm1 in elu

majnemer committed 6 years ago

d0d324d2

Merge pull request #1204 from majnemer/logaddexp

mattjj committed 6 years ago

Verified 61e52ab2

Use a faster, numerically more faithful, approach to logaddexp

majnemer committed 6 years ago

18beaab2

Merge pull request #1202 from brianwa84/patch-5

mattjj committed 6 years ago

Verified 3cca69cf

Merge pull request #1201 from brianwa84/patch-4

mattjj committed 6 years ago

Verified 7b728c6e

Merge pull request #1200 from majnemer/log1p

hawkinsp committed 6 years ago

Verified 5c77b6e2

Makes the `Tracer` object `weakref`-able

brianwa84 committed 6 years ago

Verified d07107af

Fix an exception caused by `cached()` hashing

brianwa84 committed 6 years ago

Verified 85e5d634

Use log1p when computing log(1 + x) or log(1 - x)

majnemer committed 6 years ago

1dbdaab7

Merge pull request #1198 from brianwa84/patch-3

mattjj committed 6 years ago

Verified b4ae7252

Merge pull request #1199 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 976d165e

hawkinsp committed 6 years ago

784ed8d4

Avoid a TypeError when reps is or contains a ndarray

brianwa84 committed 6 years ago

Verified 48f64ec8

Merge pull request #1193 from hawkinsp/axis

hawkinsp committed 6 years ago

Verified 9fcf5266

Merge pull request #1192 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 2160b560

Fix Python 2 compatibility problem.

hawkinsp committed 6 years ago

f4fde047

Use _canonicalize_axis everywhere to canonicalize axes, rather than sometimes mod or %.

hawkinsp committed 6 years ago

efe98e2b

Consistently return JAX arrays instead of Numpy-classic arrays from jax.numpy.

hawkinsp committed 6 years ago

1ba13e1b

Merge pull request #1191 from hawkinsp/master

hawkinsp committed 6 years ago

Verified 63c98e79

Merge pull request #1189 from hawkinsp/random

hawkinsp committed 6 years ago

Verified c138091a

Use `select` instead of `rem` to handle index wraparound.

hawkinsp committed 6 years ago

6d357fe8

Newer Older