A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations. Uses jax=0.2.29 and jaxlib 0.1.77
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool