lib/python3.11/site-packages/functorch/_src/make_functional/__init__.py

# This file has moved to under torch/_functorch. It is not public API.
# If you are not a PyTorch developer and you are relying on the following
# imports, please file an issue.
from torch._functorch.make_functional import _swap_state