ROCm/jax

Pull Requests Commits

progress w/ skye and roy

mattjj committed 6 years ago

57c1e4e6

mattjj committed 6 years ago

7ed8ec00

Merge pull request #807 from google/issue806

mattjj committed 6 years ago

Verified 14a283ae

Merge pull request #799 from google/ppermute-grad

mattjj committed 6 years ago

Verified c62e334a

namedtuple subclass transparency (fixes #806)

mattjj committed 6 years ago

fadd18b3

Merge pull request #800 from google/issue798

mattjj committed 6 years ago

Verified 9dfe2788

add jax.eval_shape to reference docs via jax.rst

mattjj committed 6 years ago

ffec059f

fix duck typing in jax.eval_shape (cf. #798)

mattjj committed 6 years ago

dda95df5

add jax.eval_shape, fixes #798

mattjj committed 6 years ago

11c512a1

improve disable_jit docstring

mattjj committed 6 years ago

93e11433

Add domain test to atanh implementation.

hawkinsp committed 6 years ago

e63bd4d5

add another ppermute test, remove extra print call

mattjj committed 6 years ago

dffe4915

ppermute (aka collective_permute) jvp + transpose

mattjj committed 6 years ago

8377224b

Merge pull request #797 from hawkinsp/prefetch

hawkinsp committed 6 years ago

Verified 0bb58847

Add recursive _prefetch_jaxpr_literals method, use it in both xla and pxla.

hawkinsp committed 6 years ago

b532e068

Incorporate review comments.

hawkinsp committed 6 years ago

6683846a

Add support for prefetching buffers from the device to the host.

hawkinsp committed 6 years ago

28e426a2

abstract_eval_fun should instantiate output pval

mattjj committed 6 years ago

0a8810d9

hawkinsp committed 6 years ago

d0415f04

Remove unused unzip2 import.

hawkinsp committed 6 years ago

70dd95bf

Merge pull request #784 from JasperSnoek/master

mattjj committed 6 years ago

Verified 7950149c

Added a note about squared distances

JasperSnoek committed 6 years ago

06d41fba

Merge pull request #792 from hawkinsp/delete

hawkinsp committed 6 years ago

Verified c598c801

Merge pull request #791 from hawkinsp/master

hawkinsp committed 6 years ago

Verified a46199cf

Remove a couple of jax.scipy.misc mentions in Colab notebooks.

hawkinsp committed 6 years ago

d0d9b8e9

Add an explicit DeviceArray.delete method that deletes both the device and host parts of a DeviceArray.

hawkinsp committed 6 years ago

af67447f

Remove jax.scipy.misc.

hawkinsp committed 6 years ago

2b235e7f

Merge pull request #789 from google/pmap-environments

mattjj committed 6 years ago

Verified 6a59a567

Make pmap lax.psum(1, 'i') and pxla.axis_index('i') work

mattjj committed 6 years ago

25262edb

Relax Bazel maximum version check.

hawkinsp committed 6 years ago

fafe69bd

Older