GitHub repository: microsoft/folx
Primary language: Python
Description: Implementation of Forward Laplacian algorithm in JAX