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