ROCm/jax

Pull Requests Commits

improve error message for #2554

mattjj committed 5 years ago

949d4231

don't hardcode array size in test

mattjj committed 5 years ago

29c581bd

Update changelog and README for jaxlib 0.1.43. (#2556)

hawkinsp committed 5 years ago

Verified bd1708c7

Disable test_while_loop_of_pmap on all platforms

gnecula committed 5 years ago

59ed4aea

Fix import in benchmarks

gnecula committed 5 years ago

fd52fbf4

fix link in custom derivatives tutorial notebook

mattjj committed 5 years ago

a4ceae1c

fix typo in notebook

mattjj committed 5 years ago

27604c39

mattjj committed 5 years ago

e017a923

add sphinx-autodoc-typehints to travis install

mattjj committed 5 years ago

7a9c550e

update api.py docstrings for sphinx highlighting

mattjj committed 5 years ago

e7be43da

Merge pull request #2551 from google/enable-mpc-tests

mattjj committed 5 years ago

Verified cda505a2

try adding sphinx-autodoc-typehints

mattjj committed 5 years ago

909fee6a

try re-enabling control tests that trigger #2507

mattjj committed 5 years ago

b015e571

try resetting global tracer state in loops_test.py

mattjj committed 5 years ago

1f03d48c

Merge pull request #2550 from google/update-custom-derivatives-notebook

mattjj committed 5 years ago

Verified de37eae6

add docstring for defjvps, fix sphinx docs

mattjj committed 5 years ago

15009c90

update custom derivatives tutorial notebook

mattjj committed 5 years ago

bd726fcd

Merge pull request #2548 from google/while-loop-of-pmap-bug

mattjj committed 5 years ago

Verified b7d2cbdc

skip new test on cpu unless num_devices > 1

mattjj committed 5 years ago

375575a5

fix a while-loop-of-pmap bug (thanks @jaspersnoek)

mattjj committed 5 years ago

df3b5fe4

Update XLA to fix CUDA 9.2 build problem. (#2547)

hawkinsp committed 5 years ago

Verified acb24146

Update XLA and increment jaxlib version to 0.1.43. (#2546)

hawkinsp committed 5 years ago

Verified 0c7e134c

don't pass `-n 1` to pytest in travis

mattjj committed 5 years ago

1275bc5d

minor fix to custom_transforms

mattjj committed 5 years ago

b43051e4

Merge pull request #2543 from google/process-custom-jvp-default-implementation

mattjj committed 5 years ago

Verified 3d0c1871

Merge pull request #2541 from google/xla-computation-duck-typing

mattjj committed 5 years ago

Verified ea9401d8

comments/defaults for process_custom_{jv,vj}p_call

mattjj committed 5 years ago

70a3f47b

allow duck-typing in xla_computation arguments

mattjj committed 5 years ago

f766c5e7

add initial_style_staging to custom_transforms

mattjj committed 5 years ago

9d8823c9

add some stage_out=True indicators

mattjj committed 5 years ago

a6a837a6

Older