functorch is a prototype of JAX-like composable function transforms for PyTorch.
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool