ROCm/jax

Pull Requests Commits

adding div and linear prims (#2338)

jessebett committed 5 years ago

Verified 4f60ed22

add more jet rules!

mattjj committed 5 years ago

cc10a12f

more jet rules and tests

mattjj committed 5 years ago

33fdb8d4

new jet implementation, with conv-based rules

mattjj committed 5 years ago

fe046f57

Added the optix.py documentation to RTD (#2312)

gnecula committed 5 years ago

Verified b21e7530

Merge pull request #2266 from google/lqr-bench

froystig committed 5 years ago

Verified 276c79e9

fix comment typo in MPC/LQR example

froystig committed 5 years ago

6da6df0a

Added .pytype cache to .gitignore (#2309)

gnecula committed 5 years ago

Verified 80630c03

Minor dedupe xla/pxla tracing code. (#2258)

chr1sj0nes committed 5 years ago

Verified 8028e4b5

Fix abstract evaluation rule for lax.top_k. (#2290)

hawkinsp committed 5 years ago

Verified 0416d2a5

JIT differentiate -> JIT compile (#2279)

joaogui1 committed 5 years ago

Verified f6e1d01f

Use int32 counters in optix. (#2239)

tomhennigan committed 5 years ago

Verified cd5bcb36

Moved CHANGELOG to docs (#2252)

gnecula committed 5 years ago

Verified 89514f92

Minor fixes to docs related to jax.numpy.vectorize (#2278)

shoyer committed 5 years ago

Verified 48f2a414

Shrink nested pmap image in demo notebook. (#2293)

skye committed 5 years ago

Verified f6d48373

Add nested pmap image to demo notebook. (#2292)

skye committed 5 years ago

Verified 7885b1d0

Unbreak build and update XLA. (#2289)

hawkinsp committed 5 years ago

Verified 80abdf0c

Always jit scipy.ndimage.map_coordinates (#2286)

shoyer committed 5 years ago

Verified 8c3e3b2d

Add an experimental lax.top_k operator. (#2280)

hawkinsp committed 5 years ago

Verified af0967fd

tweak readme pmap imports (#2276)

brettkoonce committed 5 years ago

Verified 8372a700

Add a jit around lax_linalg.lu_pivots_to_permutation (#2277)

shoyer committed 5 years ago

Verified 218a1711

remove unused imports in MPC/LQR example

froystig committed 5 years ago

9cb8171f

Update WORKSPACE to pick up TF grpc upgrade and cleanup.

skye committed 5 years ago

0a78c8c3

Revert "add np.copy method to abstract arrays (#2257)" (#2272)

gnecula committed 5 years ago

Verified 6e4ea4f7

Disable linalg_test:testMatrix power on TPU (#2269)

gnecula committed 5 years ago

Verified 4978e3c2

Add "edge" support for pad (#2265)

dalexander committed 5 years ago

Verified b995787c

fix unit handling in vmap of cond, fixes #2264 (#2268)

mattjj committed 5 years ago

Verified ab327aca

set tolerances and toggle dtype checks in MPC/LQR example tests

froystig committed 5 years ago

f02578ea

Update WORKSPACE to setup upstream dependencies. (#2267)

skye committed 5 years ago

Verified cfb5666a

move MPC/LQR example to examples directory

froystig committed 5 years ago

f1af9893

Older