ROCm/jax

Pull Requests Commits

Add logging for pmap input (re)sharding

jekbradbury committed 5 years ago

Verified d79f48b5

Merge pull request #4803 from AdrienCorenflos:patch-1

a-googler committed 5 years ago

65a7f608

Merge pull request #4854 from j-towns:tidy-stack

a-googler committed 5 years ago

83a38f4f

add correct annotations to core.TraceStack

mattjj committed 5 years ago

8b006f6a

Merge pull request #4886 from skye:jaxlib_fixes2

a-googler committed 5 years ago

18120020

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

jaxlib: Fix Python 3.9 build and drop CUDA 10.0 support

skye committed 5 years ago

5c89170d

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

Older