mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
e6ec0efaf8
Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/106205 Approved by: https://github.com/albanD
7 lines
274 B
Python
7 lines
274 B
Python
# PyTorch forward-mode is not mature yet
|
|
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
|
|
from torch._functorch.eager_transforms import hessian, jacfwd, jvp
|
|
from torch._functorch.vmap import chunk_vmap
|
|
|
|
from functorch import functionalize
|