GitHub repository: lessw2020/functorch
Description: functorch is JAX-like composable function transforms for PyTorch.