|
|
58ee527d46
|
[functorch] Added batching rules for some in-place pointwise ops
|
2022-07-21 13:40:51 -07:00 |
|
|
|
66918c4a3e
|
[functorch] Refactory UnaryOps
|
2022-07-21 13:40:51 -07:00 |
|
|
|
cb51b16e59
|
[functorch] Batching rules for comparison ops
|
2022-07-21 13:40:51 -07:00 |
|
|
|
f92fbeef74
|
[functorch] vjp now supports pytree inputs and outputs
|
2022-07-21 13:40:51 -07:00 |
|
|
|
9490aa0b65
|
[functorch] Fix vjp and grad for unrelated outputs
Fixes pytorch/functorch#1.
|
2022-07-21 13:40:51 -07:00 |
|
|
|
ec99d21d1e
|
[functorch] link colab with install instructions
|
2022-07-21 13:40:51 -07:00 |
|
|
|
9a81203259
|
[functorch] Use functorch._C._set_vmap_fallback_warning_enabled
|
2022-07-21 13:40:51 -07:00 |
|
|
|
15ab42ce7c
|
[functorch] Beef up make_functional docstring, update some examples
|
2022-07-21 13:40:51 -07:00 |
|
|
|
6ecf169a07
|
[functorch] fixed some python key tracing issues
|
2022-07-21 13:40:51 -07:00 |
|
|
|
7224611cd9
|
[functorch] Removed some stupid stuff
|
2022-07-21 13:40:51 -07:00 |
|
|
|
80131b937d
|
[functorch] adds support for grad if inputs don't depend on output (pytorch/functorch#6)
* adds support for grad if inputs don't depend on output
* fix some issues
* responded to comments
|
2022-07-21 13:40:51 -07:00 |
|
|
|
86e49cf0d7
|
[functorch] fix vmap tests and added dot shape check
|
2022-07-21 13:40:51 -07:00 |
|
|
|
73f57a7192
|
[functorch] removed nnc_compile
|
2022-07-21 13:40:51 -07:00 |
|
|
|
98e995c467
|
[functorch] Added python key stuff that builds with master
|
2022-07-21 13:40:51 -07:00 |
|
|
|
9d36895a83
|
[functorch] merged
|
2022-07-21 13:40:51 -07:00 |
|
|
|
98df806b95
|
[functorch] Fix up the cifar10_transforms example
|
2022-07-21 13:40:51 -07:00 |
|
|
|
dfeb7898f2
|
[functorch] pytree output support for vmap
|
2022-07-21 13:40:51 -07:00 |
|
|
|
c6773c67d6
|
[functorch] Add pytree support for grad and grad_and_value
|
2022-07-21 13:40:51 -07:00 |
|
|
|
8c8685ca39
|
[functorch] Fix build
|
2022-07-21 13:40:51 -07:00 |
|
|
|
ba91512ce9
|
[functorch] Installation instructions
|
2022-07-21 13:40:51 -07:00 |
|
|
|
32240466a4
|
[functorch] Added ones_like/zeros_like/... batching rules
|
2022-07-21 13:40:51 -07:00 |
|
|
|
03550ccd61
|
[functorch] Added debug commands
|
2022-07-21 13:40:51 -07:00 |
|
|
|
033e6e30af
|
[functorch] removed python key stuff
|
2022-07-21 13:40:51 -07:00 |
|
|
|
2467773967
|
[functorch] Quick temporary fix for python key
|
2022-07-21 13:40:51 -07:00 |
|
|
|
c521cd1f45
|
[functorch] fixed dot batching rule with no batch dims
|
2022-07-21 13:40:51 -07:00 |
|
|
|
0d94ae66a7
|
[functorch] fixed dot batching rule and added vmap tests using opinfo
|
2022-07-21 13:40:50 -07:00 |
|
|
|
918ede7a85
|
[functorch] Added dot/add.Scalar/mul.Scalar/etc. batching rules and added functools.wraps to grad
|
2022-07-21 13:40:50 -07:00 |
|
|
|
ac9be17a87
|
[functorch] modified example
|
2022-07-21 13:40:50 -07:00 |
|
|
|
c24314c09b
|
[functorch] More batch rules
|
2022-07-21 13:40:50 -07:00 |
|
|
|
d7d266f51e
|
[functorch] Grab bag of batch rules
|
2022-07-21 13:40:50 -07:00 |
|
|
|
0abba43aa3
|
[functorch] fix testing warning, update readme
|
2022-07-21 13:40:50 -07:00 |
|
|
|
42843306ab
|
[functorch] Add codegen file; beef up testing
|
2022-07-21 13:40:50 -07:00 |
|
|
|
e8abf483ea
|
[functorch] Run all correctness tests
|
2022-07-21 13:40:50 -07:00 |
|
|
|
20fac9da6e
|
[functorch] testing, version.txt
|
2022-07-21 13:40:50 -07:00 |
|
|
|
8277d74e42
|
[functorch] Update readme, fix unsqueeze batch rule
|
2022-07-21 13:40:50 -07:00 |
|
|
|
7001a2f1e4
|
[functorch] Update README, setup.py
|
2022-07-21 13:40:50 -07:00 |
|
|
|
70925e47c7
|
[functorch] Update
|
2022-07-21 13:40:50 -07:00 |
|
|
|
b7096ab83a
|
[functorch] misc cleanup
|
2022-07-21 13:40:50 -07:00 |
|
|
|
ce453d449e
|
[functorch] Updated some batching rules to use new API
|
2022-07-21 13:40:50 -07:00 |
|
|
|
985b35c23d
|
[functorch] new_blah hack
|
2022-07-21 13:40:50 -07:00 |
|
|
|
8201dfc2d5
|
[functorch] Various bug fixes; new BR api
|
2022-07-21 13:40:50 -07:00 |
|
|
|
a39cb7a80f
|
[functorch] beef testing
|
2022-07-21 13:40:50 -07:00 |
|
|
|
93888a3779
|
[functorch] a lot of files
|
2022-07-21 13:40:50 -07:00 |
|
|
|
608a932c1a
|
[functorch] Initial commit
|
2022-07-21 13:40:50 -07:00 |
|
