ROCm/jax

Pull Requests Commits

mention static_argnum values should be immutable

mattjj committed 5 years ago

1da36943

Merge pull request #4716 from gnecula:update_changelog

a-googler committed 5 years ago

38ba3471

Update CHANGELOG

gnecula committed 5 years ago

da9a64ef

Merge pull request #4713 from google:simpler-select-jvp

a-googler committed 5 years ago

9092ad3d

simplify select jvp

mattjj committed 5 years ago

7a73e99e

Merge pull request #4708 from google:docfix-jit

a-googler committed 5 years ago

ab3a4c2f

Merge pull request #4695 from Joshuaalbert:master

a-googler committed 5 years ago

a4956695

all changes plus test verifcation on TPU squashed

Joshuaalbert committed 5 years ago

231168d4

Merge pull request #4707 from google:traceback-register

a-googler committed 5 years ago

25b40702

Merge pull request #4709 from google:update-pypi

a-googler committed 5 years ago

4c9545d6

Merge pull request #4702 from fabianp:patch-3

a-googler committed 5 years ago

3bb0c49b

update version and changelog for pypi

mattjj committed 5 years ago

4ddfffff

fix jit docstring

froystig committed 5 years ago

9fc53b6e

move `wraps` from `api_util` to `util` to avoid cyclic dependencies

froystig committed 5 years ago

fa1d7ab5

add path exclusion opt-in to filtered stack traces and use it throughout the codebase

froystig committed 5 years ago

5d50e193

Merge pull request #4698 from google:custom-gradient-wrapper

a-googler committed 5 years ago

abbd8270

FIX: example usage of optimizers

fabianp committed 5 years ago

Verified 5d736286

add jax.custom_gradient wrapper for jax.custom_vjp

mattjj committed 5 years ago

3c6cdcfc

Merge pull request #4700 from hawkinsp:fft

a-googler committed 5 years ago

602c7836

Add PocketFFT mirror to fix download problems for some Bazel versions.

hawkinsp committed 5 years ago

9b7fd632

[JAX] Use PocketFFT for FFTs on CPU instead of Eigen.

hawkinsp committed 5 years ago

f58f1ee4

Merge pull request #4661 from jblespiau:changelist/338124676

a-googler committed 5 years ago

8121255d

Emit a warning on non-hashble static arguments in jax.jit.

jblespiau committed 5 years ago

98ee69ba

Copybara import of the project:

mattjj committed 5 years ago

9ba28d26

Copybara import of the project:

mattjj committed 5 years ago

fcaced32

Merge pull request #4658 from jakevdp:custom-obj-avals

a-googler committed 5 years ago

2677c04f

Merge pull request #4686 from google:lazy-simplification-again

a-googler committed 5 years ago

8ff63961

Merge pull request #4685 from google:make-dynamic-slice-transpose-handle-symbolic-zero-cotangents

a-googler committed 5 years ago

fe917282

Merge pull request #4684 from jakevdp:cleanup

a-googler committed 5 years ago

9845cced

redo #4535 lazy simplification

mattjj committed 5 years ago

ced333d1

Older