ROCm/jax

Pull Requests Commits

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

jblespiau committed 5 years ago

3b2018de

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

Merge pull request #4470 from gnecula:tf_tracers

a-googler committed 5 years ago

fc69d455

Fix extraneous dtype warning in jnp.mean

jakevdp committed 5 years ago

67a77c7e

[jax2tf] Port jax2tf to use omnistaging

gnecula committed 5 years ago

0213efdf

Merge pull request #4518 from gnecula:tf_debug

a-googler committed 5 years ago

e194dff6

[jax2tf] Fix the behavior for shift when the shift amount is negative or large.

gnecula committed 5 years ago

a1e67be7

Merge pull request #4517 from google:issue4510

a-googler committed 5 years ago

e6399cd7

make pytype happy

mattjj committed 5 years ago

7691be92

optimize scan partial_eval to fix #4510

mattjj committed 5 years ago

52fe026c

Merge pull request #4514 from google:remove-unneeded-code

a-googler committed 5 years ago

d4da9cc1

Merge pull request #4515 from google:enable-conv-test

a-googler committed 5 years ago

ab462baf

Merge pull request #4512 from jakevdp:raise-to-shaped

a-googler committed 5 years ago

4994b82b

re-enable lax autodiff test

froystig committed 5 years ago

1614e283

remove mysterious code that is no longer needed

mattjj committed 5 years ago

55be9dd9

Older