You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
6 lines
274 B
6 lines
274 B
4 days ago
|
# PyTorch forward-mode is not mature yet
|
||
|
|
from torch._functorch.apis import chunk_vmap
|
||
|
|
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
|
||
|
|
from torch._functorch.eager_transforms import hessian, jacfwd, jvp
|
||
|
|
from torch.func import functionalize
|