functorch (pypi): JAX-like composable function transforms for PyTorch
Keywords: gradients|hessians|pytorch