ROCm/jax

Pull Requests Commits

Print doc (#3032)

gnecula committed 5 years ago

Verified 574db2a5

try adding sphinx-autodoc-typehints

mattjj committed 5 years ago

909fee6a

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

Merge pull request #2539 from google/remat-fix

mattjj committed 5 years ago

Verified fab66007

remove remat context check, add initial staging

mattjj committed 5 years ago

bdc0c3bf

Merge pull request #2500 from google/custom-jvp-fix

mattjj committed 5 years ago

Verified e7f8503c

skip ode test on import error (internal)

mattjj committed 5 years ago

74d358d0

fix bugs, add tests

mattjj committed 5 years ago

b2b68e54

use custom_jvp for internal functions

mattjj committed 5 years ago

7a4c4d55

revamp custom_jvp/vjp implementation to fix bugs

mattjj committed 5 years ago

6193e5e4

Merge pull request #2537 from duvenaud/morejets

mattjj committed 5 years ago

Verified 954334cb

Merge pull request #2538 from google/pfau

mattjj committed 5 years ago

Verified 0f11f9c1

workaround for pmap output PRED arrays on cpu/gpu

mattjj committed 5 years ago

1762a865

Merge pull request #2536 from google/issue2534

mattjj committed 5 years ago

Verified 305dd8c2

Older