GitHub repository: onlyrico/functorch
Primary language: C++
Description: functorch is a prototype of JAX-like composable function transforms for PyTorch.