1# Copyright (c) Facebook, Inc. and its affiliates. 2# All rights reserved. 3# 4# This source code is licensed under the BSD-style license found in the 5# LICENSE file in the root directory of this source tree. 6import torch 7from torch._functorch.deprecated import ( 8 combine_state_for_ensemble, 9 functionalize, 10 grad, 11 grad_and_value, 12 hessian, 13 jacfwd, 14 jacrev, 15 jvp, 16 make_functional, 17 make_functional_with_buffers, 18 vjp, 19 vmap, 20) 21 22# utilities. Maybe these should go in their own namespace in the future? 23from torch._functorch.make_functional import ( 24 FunctionalModule, 25 FunctionalModuleWithBuffers, 26) 27 28# Was never documented 29from torch._functorch.python_key import make_fx 30 31 32# Top-level APIs. Please think carefully before adding something to the 33# top-level namespace: 34# - private helper functions should go into torch._functorch 35# - very experimental things should go into functorch.experimental 36# - compilation related things should go into functorch.compile 37 38 39__version__ = torch.__version__ 40