ROCm/jax

Pull Requests Commits

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

mattjj committed 5 years ago

32ffb247

mattjj committed 5 years ago

d5c94082

fix a custom_jvp vmap bug from @dpfau

mattjj committed 5 years ago

a46d0028

Merge pull request #4663 from google:improve-tracer-error-2

a-googler committed 5 years ago

dcfa7644

only run test with omnistaging on

mattjj committed 5 years ago

2b62a447

mattjj committed 5 years ago

49b1ec22

Merge pull request #4664 from google:4008-fixes

a-googler committed 5 years ago

8684da4c

fixes on #4008 (thanks @apaszke)

mattjj committed 5 years ago

79e3db55

Newer Older