ROCm/jax

Pull Requests Commits

Fallback when faced with other runtimes.

jblespiau committed 5 years ago

cfd65634

Merge pull request #4502 from hawkinsp:linalg

a-googler committed 5 years ago

145ac406

Merge pull request #4500 from LenaMartens:changelist/336082045

a-googler committed 5 years ago

684a5849

Recast int/bool tangents to float0 in custom_jvp/vjps

LenaMartens committed 5 years ago

e3d622cc

Merge pull request #4501 from gnecula:tf_tests_cf1

a-googler committed 5 years ago

95c0fea5

Enable TPU linalg tests that now pass.

hawkinsp committed 5 years ago

2fca9170

[jax2tf] Ensure that in tests TF does not constant-fold in eager mode before compiling

gnecula committed 5 years ago

dc9168ba

Merge pull request #4475 from bchetioui:fix_static_tests

a-googler committed 5 years ago

8df116bd

Fix failing tests on TPU by avoiding using scalars

bchetioui committed 5 years ago

82995980

Merge pull request #4491 from google:remove-test

a-googler committed 5 years ago

83060cc1

Merge pull request #4489 from google:test-fixes

a-googler committed 5 years ago

e65115b8

remove an always-skipped test, redundant by now

froystig committed 5 years ago

30b8cc57

Merge pull request #4485 from hawkinsp:nan

a-googler committed 5 years ago

4b65f33f

remove stale version guard in test

froystig committed 5 years ago

1fb097bd

skip test that is broken at unreleased (source) jaxlib

froystig committed 5 years ago

16cd3304

Exend threading.local.

jblespiau committed 5 years ago

7724322d

Switch implementation of jnp.isnan(x) to x != x.

hawkinsp committed 5 years ago

a8b67ae4

Merge pull request #4482 from google:tb-util-path

a-googler committed 5 years ago

9cdb139e

Increase the version of jaxlib, to get the tests for the C++ jit running internally.

jblespiau committed 5 years ago

4ae3352e

Merge pull request #4479 from jblespiau:changelist/335912189

a-googler committed 5 years ago

a306c1b4

Ignore an incorrect mypy error.

jblespiau committed 5 years ago

ab7436a2

early return for non-existent path prefixes

froystig committed 5 years ago

32ac909d

Use the Python jit for the compilation in the C++ jit.

jblespiau committed 5 years ago

c88be879

[jax2tf] Fix min_max and add_mul harnesses.

bchetioui committed 5 years ago

6a4fdc97

Merge pull request #4468 from gnecula:tf_colab4

a-googler committed 5 years ago

0290339e

Another attempt to save the notebook

gnecula committed 5 years ago

ae7e5b99

Updated the notebook with a run on a public runtime

gnecula committed 5 years ago

27e5c2a7

gnecula committed 5 years ago

50ce9150

[jax2tf] Add the getting started notebook

gnecula committed 5 years ago

ed0a098d

Merge pull request #4467 from google:update-pypi

a-googler committed 5 years ago

2dfc5f96

Older