ROCm/jax

Pull Requests Commits

mattjj committed 6 years ago

55e368ae

mattjj committed 6 years ago

6b4a806d

brainstorming prng linearity checks, PAIR=dougalm

mattjj committed 6 years ago

35aa236f

version bump for pypi

mattjj committed 6 years ago

76decd2a

Merge pull request #313 from google/skip-some-tests

mattjj committed 6 years ago

Verified aa5c375d

skip some cases to satisfy internal tests

mattjj committed 6 years ago

9846c9c7

add more special cases of select batching rule

mattjj committed 6 years ago

bf7a438c

Merge pull request #310 from google/issue292

mattjj committed 6 years ago

Verified 44cffd00

Merge pull request #312 from google/issue311

mattjj committed 6 years ago

Verified dc1939f2

add an efficient special case to select batch rule

mattjj committed 6 years ago

583b6547

add lax.select broadcasting tests, improve rule

mattjj committed 6 years ago

5344e7ae

generalize select batch rule (fixes #311)

mattjj committed 6 years ago

fe96c15d

improve error messages for lax.slice/index funs

mattjj committed 6 years ago

0afb6202

Merge pull request #308 from google/pjit

mattjj committed 6 years ago

Verified 1ab4a2ea

fix python3 bug

mattjj committed 6 years ago

901e221c

remove pjit_model.py for merge

mattjj committed 6 years ago

72999172

fix broken test

mattjj committed 6 years ago

2801b364

revise parallel model file

mattjj committed 6 years ago

7d1527b4

the many ways to pjit a cat

mattjj committed 6 years ago

b16b4ddf

make pjit out_axes=None work

mattjj committed 6 years ago

a8c27681

Merge branch 'master' into pjit

mattjj committed 6 years ago

055beb90

delete more dead index_take code

mattjj committed 6 years ago

f5cffd72

Merge pull request #307 from google/use-gather-for-indexing

mattjj committed 6 years ago

Verified 7ea2bd72

index_take in terms of gather, delete index_untake

mattjj committed 6 years ago

9f3060a0

Add `np.take` to the documentation index.

hawkinsp committed 6 years ago

fec5f475

Merge pull request #306 from hawkinsp/take

hawkinsp committed 6 years ago

Verified 9480c983

Implement np.take (#70).

hawkinsp committed 6 years ago

eb80ef68

fix merging issue

mattjj committed 6 years ago

f69dda96

remove stale OutAxesThunk class

mattjj committed 6 years ago

9a251efd

move basic pjit+autodiff tests

mattjj committed 6 years ago

092712fd

Older