blob: 8d9133606c5299e40e10c4c4f1dfde10ae4222b2 [file] [log] [blame]
from torch._functorch.eager_transforms import (
grad,
grad_and_value,
vjp,
jvp,
jacrev,
jacfwd,
hessian,
functionalize,
)
from torch._functorch.vmap import vmap