ROCm/jax

Pull Requests Commits

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

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

Merge pull request #4681 from jblespiau:changelist/338538921

a-googler committed 5 years ago

a8b90566

make lax.dynamic_slice transpose handle symb zeros

mattjj committed 5 years ago

f40ac067

Cleanup: pass function name rather than function object

jakevdp committed 5 years ago

2af475d9

Fix a bug in `make_jaxpr`, as static-arguments should be passed as is

jblespiau committed 5 years ago

225cdb98

[multi-buf] simplify custom object test avals

jakevdp committed 5 years ago

48db25e6

Merge pull request #4644 from jblespiau:changelist/337949790

a-googler committed 5 years ago

20ee4150

Raise a meaningful exception in the C++ jax.jit on non-hashable object static argnum.

jblespiau committed 5 years ago

62cbc231

Merge pull request #4635 from aldragan0:numpy-setdiff1d

a-googler committed 5 years ago

cbd07798

[impl] Add support for setdiff1d

aldragan0 committed 5 years ago

b7fee456

Merge pull request #4459 from romanngg:extract_patches

a-googler committed 5 years ago

72a273f7

Merge pull request #4650 from gnecula:abstract_check

a-googler committed 5 years ago

bcc1d71a

Add `lax.conv_general_dilated_patches`

romanngg committed 5 years ago

da0bff2f

Update jax/core.py

gnecula committed 5 years ago

Verified c5b983c0

Merge pull request #4665 from google:fix-custom-jvp-vmap-bug

a-googler committed 5 years ago

e098fbda

Older