All modules for which code is available functorch._src.aot_autograd functorch._src.compilers functorch._src.eager_transforms functorch._src.make_functional functorch._src.partitioners functorch._src.vmap