pytorch/functorch

functorch is JAX-like composable function transforms for PyTorch.

语言：Jupyter Notebook · GitHub 星标：1,433 · Fork：108

话题：gradients · hessians · pytorch

在 GitHub 上查看原仓库