mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
26f4f10ac8
The `usort` config in `pyproject.toml` has no effect due to a typo. Fixing the typo make `usort` do more and generate the changes in the PR. Except `pyproject.toml`, all changes are generated by `lintrunner -a --take UFMT --all-files`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127126 Approved by: https://github.com/kit1980
6 lines
273 B
Python
6 lines
273 B
Python
# PyTorch forward-mode is not mature yet
|
|
from functorch import functionalize
|
|
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
|