ROCm/jax

Pull Requests Commits

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

mattjj committed 5 years ago

9b5d7806

fix a 'store occupied' error in jax_debug_nans

mattjj committed 5 years ago

c26c77d2

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

update version and changelog for pypi

mattjj committed 5 years ago

1c5a09ac

Merge pull request #4464 from google:minor-source-sync

a-googler committed 5 years ago

cf1fefcd

minor change as source sync test

froystig committed 5 years ago

5f6fab8d

Internal change

a-googler committed 5 years ago

6eb4f54e

Revert "[jax2tf] Added Getting Started notebook (#4460)" (#4463)

gnecula committed 5 years ago

Verified 93f4dd44

[jax2tf] Added Getting Started notebook (#4460)

gnecula committed 5 years ago

Verified 1d739527

Merge pull request #4457 from google:tycheck-invar-eqn-context

a-googler committed 5 years ago

2166f0ee

Merge pull request #4458 from hawkinsp:xla

a-googler committed 5 years ago

7efa3209

hawkinsp committed 5 years ago

49d851d2

equation context for undefined var reads in jaxpr typechecker

froystig committed 5 years ago

e7979258

Merge pull request #4456 from jakevdp:small-fix

a-googler committed 5 years ago

91755e5c

Merge pull request #4455 from johnpjf:patch-2

a-googler committed 5 years ago

9273e8c4

Merge pull request #4444 from jakevdp:dtype-check

a-googler committed 5 years ago

efeed6c1

Newer Older