GitHub repository: yushangdi/functorch
Primary language: Python
Description: functorch is JAX-like composable function transforms for PyTorch.