functorch is JAX-like composable function transforms for PyTorch.
Home Page:https://pytorch.org/functorch/
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool