1from torch._functorch import config 2from torch._functorch.aot_autograd import ( 3 aot_function, 4 aot_module, 5 aot_module_simplified, 6 compiled_function, 7 compiled_module, 8 get_aot_compilation_context, 9 get_aot_graph_name, 10 get_graph_being_compiled, 11 make_boxed_compiler, 12 make_boxed_func, 13) 14from torch._functorch.compilers import ( 15 debug_compile, 16 default_decompositions, 17 draw_graph_compile, 18 memory_efficient_fusion, 19 nnc_jit, 20 nop, 21 print_compile, 22 ts_compile, 23) 24from torch._functorch.fx_minifier import minifier 25from torch._functorch.partitioners import ( 26 default_partition, 27 draw_graph, 28 min_cut_rematerialization_partition, 29) 30from torch._functorch.python_key import pythonkey_decompose 31