pytorch

1
from torch._functorch.eager_transforms import (
2
    vjp,
3
    jvp,
4
    jacrev,
5
    jacfwd,
6
    hessian,
7
    functionalize,
8
    linearize
9
)
10
from torch._functorch.apis import grad, grad_and_value
11
from torch._functorch.functional_call import functional_call, stack_module_state
12
from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
13
from torch._functorch.apis import vmap
14