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