ROCm/jax

Pull Requests Commits

add trivial pjit tests

mattjj committed 6 years ago

1360a703

add post_process_call for vmap (untested)

mattjj committed 6 years ago

1a494d07

remove chunk transform from api.py for now

mattjj committed 6 years ago

32cda396

woo tuple output works

mattjj committed 6 years ago

2487eb2d

mattjj committed 6 years ago

da2d1854

mattjj committed 6 years ago

945fa34e

moving pxla flattening/chunking to api.py, wip

mattjj committed 6 years ago

780106f8

add leak_bug.py for repro

mattjj committed 6 years ago

afc2b9fe

hmmm leak bug somewhere

mattjj committed 6 years ago

a5661ce1

cleanup, fix chunksize bug

mattjj committed 6 years ago

80b9bdc4

mattjj committed 6 years ago

9c8018a4

single-axis-var version runs! and some cleanup

mattjj committed 6 years ago

da724d31

chunk transform outside of xla_parallel_callable

mattjj committed 6 years ago

8bc579ae

woo post_process_call, it scrolls like butter

mattjj committed 6 years ago

6494504a

promising... but why do we need post_process_call?

mattjj committed 6 years ago

8a6c0949

add pjit to api.py

mattjj committed 6 years ago

d321e2a3

add chunking transform

mattjj committed 6 years ago

e546f13b

add basic split test

mattjj committed 6 years ago

c1be2aa2

fix split tracer bugs

mattjj committed 6 years ago

ae86b9a6

fix pmap nesting bug, sketch out index splitting

mattjj committed 6 years ago

7e05f94c

remove 'chunk' for now

mattjj committed 6 years ago

b34c87dc

clean up parallel_test.py file

mattjj committed 6 years ago

78157722

initial pmap/pxla code, pair-coded w/ @dougalm

mattjj committed 6 years ago

c7ce4420

Add license header to docs/conf.py.

hawkinsp committed 6 years ago

326c3dbb

Merge pull request #271 from google/enable-value-and-grad-api-test

mattjj committed 6 years ago

Verified fab3a7a7

enable value-and-grad api test

mattjj committed 6 years ago

b95669eb

Merge pull request #270 from cclauss/patch-1

mattjj committed 6 years ago

Verified 8565a348

Undefined name: 'value_and_grad' in ./tests/api_test.py

cclauss committed 6 years ago

Verified f6d9a33a

Merge pull request #267 from hawkinsp/master

hawkinsp committed 6 years ago

Verified e609be4c

Update Jaxlib references to 0.1.6.

hawkinsp committed 6 years ago

1d9be64e

Older