I am done

rl/Lib/site-packages/functorch/experimental/__init__.py

@@ -0,0 +1,5 @@
+# PyTorch forward-mode is not mature yet
+from functorch import functionalize
+from torch._functorch.apis import chunk_vmap
+from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_
+from torch._functorch.eager_transforms import hessian, jacfwd, jvp

rl/Lib/site-packages/functorch/experimental/control_flow.py

@@ -0,0 +1,7 @@
+from torch import cond  # noqa: F401
+from torch._higher_order_ops.cond import UnsupportedAliasMutationException  # noqa: F401
+from torch._higher_order_ops.map import (  # noqa: F401
+    _stack_pytree,
+    _unstack_pytree,
+    map,
+)

rl/Lib/site-packages/functorch/experimental/ops.py

@@ -0,0 +1 @@
+from torch._ops import HigherOrderOperator  # noqa: F401