GitHub repository: ucl-bug/jaxdf
Primary language: Python
Description: A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations