search

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.
https://pytorch.org/functorch/
BSD 3-Clause "New" or "Revised" License
1.38k stars 102 forks source link

Remove nn.Module from aot_module_simplified #970

Closed jansel closed 1 year ago

jansel commented 1 year ago

Lower some overheads.

cc @Chillee