Warning
This version of fadge
works only with jax
and jaxlib
0.3.
But jaxlib
0.3 is not available on PyPI.
To install this version of fadge
, please first install jaxlib
with:
pip install jaxlib==0.3.25 -f https://storage.googleapis.com/jax-releases/jax_releases.html
Fast Automatic Differential GEometry (Fadge
) uses
Google's JAX
for differential geometry.