functorch is JAX-like composable function transforms for PyTorch.
Jupyter Notebook 1.44 k
2 天前
Fast non-allocating calculations of gradients, Jacobians, and Hessians with sparsity support
Julia 268
6 个月前
functorch is JAX-like composable function transforms for PyTorch.
Fast non-allocating calculations of gradients, Jacobians, and Hessians with sparsity support