ROCm/jax

Pull Requests Commits

Add some missing functions to documents. (#2615)

hawkinsp committed 5 years ago

Verified 7629c5aa

Handle n==0 case in TPU cumsum/cumprod. (#2617)

hawkinsp committed 5 years ago

Verified 36c529d4

Add backend-specific lowering for cumsum/cumprod on TPU. (#2614)

hawkinsp committed 5 years ago

Verified 329321b0

Adapted vmap out_axes check to sum_match

gnecula committed 5 years ago

f5f35c5c

Expand docstring for vmap with details about out_axes, and improve error

gnecula committed 5 years ago

7a5d1bc0

Fix lax.broadcast_shapes returning numpy ints in shape tuple (#2471)

sharadmv committed 5 years ago

Verified 99944d12

Support pytrees in jax.scipy.linalg.cg (#2600)

shoyer committed 5 years ago

Verified 1cf708ea

Merge pull request #2601 from adarob/fix-err

mattjj committed 5 years ago

Verified 2c4ced21

Fix error handling when an attempt is made to pmap a scalar.

adarob committed 5 years ago

48eb5243

add notes to changelog

mattjj committed 5 years ago

c2f56fbd

Merge pull request #2591 from google/tracer-printing

mattjj committed 5 years ago

Verified 60de46a1

Merge pull request #2599 from sharadmv/logit-fix

mattjj committed 5 years ago

Verified b67250eb

Fix dtype error

sharadmv committed 5 years ago

3c9fb35f

Implement scipy.sparse.linalg.cg (second try) (#2566)

shoyer committed 5 years ago

Verified 1b93bb51

Fix grad(logit) to use defjvps and enable it in tests

sharadmv committed 5 years ago

72783bb7

Make reduce_prod differentiable to arbitrary order. (#2597)

hawkinsp committed 5 years ago

Verified 2b3befff

Reimplement np.cumsum and np.cumprod in terms of a parallel prefix scan. (#2596)

hawkinsp committed 5 years ago

Verified 824ac866

Merge pull request #2561 from madisonmay/batch-norm-no-scale-or-center

mattjj committed 5 years ago

Verified 192e9086

Merge pull request #2563 from google/callable-typechecks

mattjj committed 5 years ago

Verified 6402419a

Merge pull request #2592 from google/travis-mypy

mattjj committed 5 years ago

Verified 25aeb803

Merge pull request #2593 from google/issue2578

mattjj committed 5 years ago

Verified 3dee689e

skip all parallelize tests (abandonware right now)

mattjj committed 5 years ago

ba8225f3

add full lower to custom_jvp/vjp call bind

mattjj committed 5 years ago

0e49133e

add trace state check tearDown to JaxTestCase

mattjj committed 5 years ago

f2de1bf3

make tracers tree-pretty-print their contents

mattjj committed 5 years ago

297c9024

Merge pull request #2587 from google/travis-mypy

mattjj committed 5 years ago

Verified 64a7d172

tell mypy: using __init__ to reinitialize is OK

mattjj committed 5 years ago

5d3f1bdf

make core.trace_state resetting be thread-local

mattjj committed 5 years ago

6d4987cc

add global trace state checks to more tests

mattjj committed 5 years ago

b78b7a03

check sublevel is reset in loops_test.py

mattjj committed 5 years ago

ab0a0054

Newer Older