GitHub repository: pytorch/functorch
Primary language: Jupyter Notebook
Description: functorch is JAX-like composable function transforms for PyTorch.