ROCm/jax

Pull Requests Commits

remove double warning with asarray dtype='int'

mattjj committed 5 years ago

dbebc9ad

Merge pull request #4548 from hawkinsp:truncnorm

a-googler committed 5 years ago

d1ca3b3d

Ensure values returned by jax.random.truncated_normal() are in range.

hawkinsp committed 5 years ago

080007ab

Merge pull request #4547 from jblespiau:changelist/336647290

a-googler committed 5 years ago

5a3929e7

Enable the C++ jax.jit fast code-path by default.

jblespiau committed 5 years ago

0660939a

Merge pull request #4554 from skye:workspace

a-googler committed 5 years ago

639cda29

Update XLA in WORKSPACE

skye committed 5 years ago

b8cac03c

Enrich the error messages with the bound names that are available.

jblespiau committed 5 years ago

b13775f4

Merge pull request #4550 from jblespiau:changelist/336694640

a-googler committed 5 years ago

7c4b935c

Copybara import of the project:

jblespiau committed 5 years ago

ee9ca569

Gate some jax_jit test with a version check.

jblespiau committed 5 years ago

5a097f5c

Add support for jax_debug_nans and fix the last few glitches with the C++ jax.jit.

jblespiau committed 5 years ago

c1e25953

Merge pull request #4546 from hawkinsp:flakes

a-googler committed 5 years ago

8f647151

Fix batching_test flakiness on GPU.

hawkinsp committed 5 years ago

e1adbcd4

Merge pull request #4440 from google:long-line

a-googler committed 5 years ago

9d8139e0

Merge pull request #4542 from google:fix-jax2tf-typo

a-googler committed 5 years ago

7320fc7c

remove trailing whitespace in jax2tf readme lines

mattjj committed 5 years ago

bcf32093

fix typo "differentiaion"

mattjj committed 5 years ago

e1708f5c

Merge pull request #4538 from google:fix-mypy

a-googler committed 5 years ago

e6ab45e0

ignore mypy error in jax2tf

mattjj committed 5 years ago

2839b2bf

Merge pull request #4509 from malmaud:pmin_multihost

a-googler committed 5 years ago

20e11a0c

Merge pull request #4532 from gnecula:tf_omnistaging

a-googler committed 5 years ago

a97cb870

[jax2tf] Ensure jax2tf still works without omnistaging

gnecula committed 5 years ago

9d9762b7

Merge pull request #4065 from skye:wheels

a-googler committed 5 years ago

0d69a84f

Use local version identifiers to distribute cuda jaxlib wheels.

skye committed 5 years ago

cacb0175

Merge pull request #4524 from jakevdp:fix-mean

a-googler committed 5 years ago

ae3d3101

Merge pull request #4526 from google:4510-2

a-googler committed 5 years ago

cd03bea5

mattjj committed 5 years ago

78fe038c

add test for scan input forwarding

mattjj committed 5 years ago

0aeeb63f

make scan fwd raw extensive inputs as DeviceArray

mattjj committed 5 years ago

5355776e

Older