GitHub repository: sail-sg/autofd
Primary language: Python
Description: Automatic Functional Differentiation in JAX