A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations. Uses jax=0.2.29 and jaxlib 0.1.77
Primary LanguagePythonGNU Lesser General Public License v3.0LGPL-3.0
No issues in this repository yet.