ROCm/jax

Pull Requests Commits

adapt broadcast_in_dim shape rule mask soundness

mattjj committed 5 years ago

8bd74a24

revise lax broadcasting to work with sound masking

mattjj committed 5 years ago

799325bb

update concatenate shape rule for sound mask test

mattjj committed 5 years ago

df081393

Fix select_and_gather_add_transpose rule.

LenaMartens committed 5 years ago

91777c33

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

qiuminxu committed 5 years ago

0f8ea375

Add named_call public API.

qiuminxu committed 5 years ago

31600aac

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

[jax2tf] Add a harness for broadcast_p and fix conversion.

bchetioui committed 5 years ago

07807d76

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

[masking] Ensure that Poly operations are sound

gnecula committed 5 years ago

740cc913

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

hawkinsp committed 5 years ago

2b8d840c

Newer Older