ROCm/jax

Pull Requests Commits

removing stray selects doesn't help much

mattjj committed 5 years ago

7c76e66d

speeding up scan compile times

mattjj committed 5 years ago

200ac8ef

Implement shapecheck for more primitives (#1990)

juliuskunze committed 5 years ago

55c971e4

Add betainc to JAX (#1998)

srvasude committed 5 years ago

80b35dd4

[pmap] Add support for nested pmaps on multihost platforms via axis_size (#2002)

trevorcai committed 5 years ago

12975bbc

Defer to unrecognized types in arithmetic (#1942)

shoyer committed 5 years ago

Verified a5644edb

Update references to bazel repositories in WORKSPACE to match TF head. (#2005)

hawkinsp committed 5 years ago

Verified 653001aa

Use a uniform rng rather than a normal rng to defeat CSE. (#2000)

hawkinsp committed 5 years ago

Verified 11224bd2

Remove :libjax alias from BUILD file. (#1996)

hawkinsp committed 5 years ago

Verified 938a7f80

added batching to LAPACK triangular_solve (#1985)

akbir committed 5 years ago

dcda87d0

Update XLA. (#1997)

hawkinsp committed 5 years ago

Verified 64bf55dc

Drop fastcache dependency, which isn't necessary on Python 3. (#1995)

hawkinsp committed 5 years ago

Verified 681ba37f

Real valued FFTs (#1657)

shoyer committed 5 years ago

Verified a5b6e8ab

Fix PmapTest.testCollectivePermuteCyclicWithPShuffle.

skye committed 5 years ago

9919fe5e

Fix PmapTest.testPShuffleWithBadPerm regexp.

skye committed 5 years ago

cfc854c6

add mini-libraries readme

mattjj committed 5 years ago

a7eb5897

fix broken link to trax, fixes #1974

mattjj committed 5 years ago

9afa2c6b

Merge pull request #1982 from noble-ai/rfftfreq

mattjj committed 5 years ago

Verified acbd2676

Added pshuffle (#1975)

Chase Roberts committed 5 years ago

34ede6b7

removed redundant comments

archis committed 5 years ago

8b6f660d

added rfftfreq, tests, and documentation link.

archis committed 5 years ago

05f09fc9

Add ppermute as an allowed multi-host collective. (#1981)

skye committed 5 years ago

Verified ed33d102

Disable failing GPU test for now pending XLA fix.

skye committed 5 years ago

160cc43a

Adjust tolerance for LaxTest.testConv0DIsDot. (#1978)

skye committed 5 years ago

Verified 773ebe13

Fix `jax.lax.axis_index` in multi-host setting. (#1976)

skye committed 5 years ago

Verified 0417e1e5

Merge pull request #1855 from JuliusKunze/categorical

mattjj committed 5 years ago

Verified 00be20bd

Require shape = sample_shape + batch_shape in random.categorical

juliuskunze committed 5 years ago

f36d858c

fix urllib import for py3

mattjj committed 5 years ago

8bca2c90

Handle 0D convolutions correctly in shape rule. (#1972)

hawkinsp committed 5 years ago

Verified facbe0d7

Merge pull request #1944 from clemisch/master

mattjj committed 5 years ago

Verified 327dca8f

Older