ROCm/jax

Pull Requests Commits

[WIP] final-style unzip for key-split-based NNs

jekbradbury committed 6 years ago

af9a0d58

experiments on stax

mattjj committed 6 years ago

8f2c83a3

add try/finally to pxla global state management

mattjj committed 6 years ago

fd9e0333

update jaxlib version for macOS wheels

mattjj committed 6 years ago

009ff074

Workaround to make XRT work with multi-backend change.

skye committed 6 years ago

Verified d5820953

reset jax_numpy_rank_promotion to "allow" default

mattjj committed 6 years ago

Verified 1cd37bd9

Merge pull request #1253 from google/custom-transforms-vjp-nones

mattjj committed 6 years ago

Verified 449e27b3

leave todos for better error messages

mattjj committed 6 years ago

dbe56c30

fix for custom-transforms vjp nones bug

mattjj committed 6 years ago

6a81d81d

fix notebook link (fixes #1252)

mattjj committed 6 years ago

94b1dda4

Add jax.devices() and friends, and add `devices` arg to pmap.

skye committed 6 years ago

ae835b74

Merge pull request #1211 from levskaya/multibackend

skye committed 6 years ago

Verified 2d26ac38

Merge pull request #1250 from fehiepsi/dtype1

mattjj committed 6 years ago

Verified 89403229

cast float64 to canonical dtype in np.cov

fehiepsi committed 6 years ago

28df8a66

update xla link, fixes #1248

mattjj committed 6 years ago

524d4254

instantiate zeros for custom vjp rules

mattjj committed 6 years ago

f639b808

tweak rank promotion warning docs

mattjj committed 6 years ago

c6f430a8

attempt to fix sphinx rst rendering

mattjj committed 6 years ago

94fafe88

improve rank promotion warning, add doc page

mattjj committed 6 years ago

3c2a7359

bump version for pypi

mattjj committed 6 years ago

e27d3c2f

Merge branch 'master' into multibackend

mattjj committed 6 years ago

Verified 0cc21c8d

Merge pull request #1246 from google/issue1241

mattjj committed 6 years ago

Verified 7a119c0c

fix device_constant instantiation bug, fixes #1241

mattjj committed 6 years ago

434d1753

Update minimum jaxlib to 0.1.26.

skye committed 6 years ago

Verified dfee03c3

Merge pull request #1234 from google/issue1230

mattjj committed 6 years ago

Verified 4838440f

address reviewer comments

mattjj committed 6 years ago

afe21baf

Bump jaxlib version to 0.1.26 and update WORKSPACE.

skye committed 6 years ago

4a22f565

add dtype warnings to array-creation routines

mattjj committed 6 years ago

e90457d7

clean up multibackend tests

levskaya committed 6 years ago

91a23116

increase minimum jaxlib version and remove some janky feature detection

levskaya committed 6 years ago

a57a1a39

Older