Fix staggered gradients

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

GNU Lesser General Public License v3.0

120 stars 7 forks source link

Closed astanziola closed 2 years ago

astanziola commented 2 years ago

Also updates the doc requirements to the new mkdocs and mkdocs-jupyter