ROCm/jax

Pull Requests Commits

Cloned from CL 333372978 by 'g4 patch'.

a-googler committed 5 years ago

14b4a6cc

Merge pull request #4890 from google:autodiff-cookbook-fix

a-googler committed 5 years ago

8358c016

fix typo in autodiff cookbook

mattjj committed 5 years ago

491ad965

Merge pull request #4887 from skye:jaxlib

a-googler committed 5 years ago

a81179f6

README etc. updates for new jaxlib release

skye committed 5 years ago

36799f50

Merge pull request #4832 from shoyer:gmres_qr

a-googler committed 5 years ago

66cf93f1

Merge pull request #4877 from jakevdp:prng-seed-invariance

a-googler committed 5 years ago

ce360e69

Relax GMRES test tolerances

shoyer committed 5 years ago

fca56663

Update CHANGELOG for PRNGKey change

jakevdp committed 5 years ago

d3aced04

Merge pull request #4871 from bchetioui:test_conj

a-googler committed 5 years ago

2b3d019c

Merge pull request #4872 from bchetioui:test_cumsum_cumprod

a-googler committed 5 years ago

13c965a7

PRNGSeed: ensure JIT invariance for valid inputs.

jakevdp committed 5 years ago

18228f45

[jax2tf] Added testing and fixed conversion of conj.

bchetioui committed 5 years ago

850e7a87

[jax2tf] Replaced cumsum/cumprod implementation and added testing.

bchetioui committed 5 years ago

5cde3cb1

Merge pull request #4875 from bchetioui:test_broadcast_in_dim

a-googler committed 5 years ago

d4b12154

[jax2tf] Fix implementation and add tests for broadcast_in_dim.

bchetioui committed 5 years ago

c5bc2ffe

Merge pull request #4879 from jakevdp:prng-output-test

a-googler committed 5 years ago

1e625dd3

TST: add explicit tests of PRNGKey seed handling.

jakevdp committed 5 years ago

297f0359

[JAX] Remove uses of the deprecated jax.experimental.vectorize.

hawkinsp committed 5 years ago

2b8d840c

Merge pull request #4820 from jakevdp:fix-type-promotion

a-googler committed 5 years ago

f1b14aa2

Merge pull request #4864 from jakevdp:doc-copy-year

a-googler committed 5 years ago

4a1fe9ea

More unit-tests + mark gmres as internal for now

shoyer committed 5 years ago

7e62270e

Merge pull request #4863 from apaszke:pmap-in-axes

a-googler committed 5 years ago

ec74c68c

DOC: update webpage copyright year to 2020

jakevdp committed 5 years ago

76dc2c1e

Merge pull request #4860 from jakevdp:cleanup

a-googler committed 5 years ago

b753f741

Add support for pmap in_axes other than 0 and None

apaszke committed 5 years ago

a5bc7353

Internal change

a-googler committed 5 years ago

bdd79156

cleanup: remove unused test arg

jakevdp committed 5 years ago

898fa7e6

Merge pull request #4848 from apaszke:pmap-in-axes

a-googler committed 5 years ago

342d62dd

Add support for pmap in_axes other than 0 and None

apaszke committed 5 years ago

6914058c

Older