ROCm/jax

Pull Requests Commits

Bump TF hash to get an upstream LLVM GCC fix.

zhangqiaorjc committed 5 years ago

b797554f

Reorder nocuda/cuda build to fail early. (#4243)

zhangqiaorjc committed 5 years ago

Verified adb34488

Update install_cuda script to specify cublas. (#4240)

zhangqiaorjc committed 5 years ago

Verified 0b04439f

Update TF dep to a passing commit hash. (#4239)

zhangqiaorjc committed 5 years ago

Verified a14133aa

Interrupt lu transformation generators whenever an exception occurs

apaszke committed 5 years ago

3f8aaabb

[jax2tf] Add a template file for documentation generation. (#4219)

bchetioui committed 5 years ago

Verified 70891f46

[jax2tf] Clean up test_dynamic_slice. (#4236)

bchetioui committed 5 years ago

Verified 053cd5aa

Add axis_index_groups support to all_gather. (#4194)

inoryy committed 5 years ago

Verified bff24bdd

[jax2tf] Updated test_pad to test all dtypes and remove old (#4235)

bchetioui committed 5 years ago

Verified f908f6f2

[jax2tf] Clean up code for XlaGather, experimental_compile not necessary (#4030)

gnecula committed 5 years ago

Verified ee38e716

Fix Dockerfile wheel installation issues. (#4232)

zhangqiaorjc committed 5 years ago

Verified 3cf73367

improve lax.pad shape rule (#4234)

mattjj committed 5 years ago

Verified 745d90d0

jaxlib build fixes. (#4066)

skye committed 5 years ago

Verified cf2d15d4

Update jaxlib version for dlpack fix. (#4231)

zhangqiaorjc committed 5 years ago

Verified 4600dd79

Add test for dtype coverage of jax.numpy ufuncs (#3913)

jakevdp committed 5 years ago

Verified 9a70be24

updtate version and changelog for pypi (#4224)

mattjj committed 5 years ago

Verified 7f3078b7

tweak lax.py shape broadcasting logic (#4217)

mattjj committed 5 years ago

Verified ed0d8c02

[jax2tf] Fix bug in population count and move expect_tf_exception (#4214)

bchetioui committed 5 years ago

Verified 798a2648

[jax2tf] Fix missing complex64 TPU corner case of scatter_{add,mul} (#4213)

bchetioui committed 5 years ago

Verified e1340f34

Add more context to the axis_frame error message.

apaszke committed 5 years ago

0aed1f4d

[jax2tf] Do not use jax.random.PRNGKey before in primitive harness (#4211)

gnecula committed 5 years ago

Verified 4413bb8a

[jax2tf] Expand coverage of primitives by categorize. (#4209)

bchetioui committed 5 years ago

Verified be8ea144

[jax2tf] Fix random.split when jax_exable_x64 (#4208)

gnecula committed 5 years ago

Verified 1e84cbe9

[jax2tf] Cleanup the correctness stats layout. (#4201)

bchetioui committed 5 years ago

Verified 6c62935d

[jax2tf] Use the JAX impl rule for threefry instead of writing our own (#4204)

gnecula committed 5 years ago

Verified c6e6ee2d

Add reverse flag in associative scan (#4181)

AdrienCorenflos committed 5 years ago

Verified 96278e67

[jax2tf] Generator for the documentation of operations with limited support (WIP) (#4193)

bchetioui committed 5 years ago

Verified bcf9777b

[jax2tf] Flip the with_gradient=True; was flipped back by mistake (#4200)

gnecula committed 5 years ago

Verified abdd1388

[jax2tf] Implementation of random_gamma (#4192)

gnecula committed 5 years ago

Verified 5eac4772

Add jax.numpy.array_split (#4197)

ajcr committed 5 years ago

Verified 708d07d5

Older