ROCm/jax

Pull Requests Commits

floating point identity to inf

Erich Elsen committed 5 years ago

491fcbb2

bump version for pypi (#3590)

mattjj committed 5 years ago

Verified 99cfbfb7

odeint: don't hoist non-differentiable consts (#3587)

mattjj committed 5 years ago

Verified 57915dc1

remove trailing whitespace

Erich Elsen committed 5 years ago

b8d0de63

consolidate tests

Erich Elsen committed 5 years ago

d7e3495d

remove now unneeded type def

Erich Elsen committed 5 years ago

290d608e

consolidate jvp rule definitions

Erich Elsen committed 5 years ago

1f15ffc4

actually return the primitive

Erich Elsen committed 5 years ago

a98249d7

add generic reducer primitive generator and replace prod/max/min with it.

Erich Elsen committed 5 years ago

a189737e

remove unit and determine automatically for all ops

Erich Elsen committed 5 years ago

d3f6d85d

fix other branch

Erich Elsen committed 5 years ago

4fe9c1d6

account for different names of reducer in tpu function

Erich Elsen committed 5 years ago

1e33e534

Also update custom tpu rule to set unit correctly based on dtype

Erich Elsen committed 5 years ago

294d6f89

Add default value of None for unit in TPU impl of cummax/cummin

Erich Elsen committed 5 years ago

a54a38f6

onp.finfo -> jnp.finfo for bfloat16

Erich Elsen committed 5 years ago

e2fa89db

use correct iinfo finfo names

Erich Elsen committed 5 years ago

ae9e6851

don't require passing identity value. It isn't the initial value - identity is required for implementation correctness

Erich Elsen committed 5 years ago

812d2462

add trailing p to excluded names

Erich Elsen committed 5 years ago

e9b1aae6

add cummax/min to __init__.py

Erich Elsen committed 5 years ago

99195b25

ignore cummax/cummin in jax2tf

Erich Elsen committed 5 years ago

80b637dc

Erich Elsen committed 5 years ago

95e15b64

numpy doesn't have a cummax, but it does have maximum.accumulate - curious api choice

Erich Elsen committed 5 years ago

156f2e81

Erich Elsen committed 5 years ago

45db78d4

Erich Elsen committed 5 years ago

bf06633a

Issue1635 expm frechet (#2062)

sriharikrishna committed 5 years ago

Verified 7b57dc8c

Incorporate a few recent NumPy API extensions. (#3586)

hawkinsp committed 5 years ago

Verified 1f2025e1

Implement DeviceArray.tobytes(). (#3585)

hawkinsp committed 5 years ago

Verified 17fc8b75

[jax2tf] Add special case for translation of lax.gather to tf.gather. (#3486)

gnecula committed 5 years ago

Verified 496cde6e

lax.sort: stable by default

froystig committed 5 years ago

ccb640af

detailed assertion in profiler test

froystig committed 5 years ago

6eea0ce1

Newer Older