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