Warning
This version of fadge
works only with jax
and jaxlib
0.3.
However, jaxlib
0.3 is no longer available on PyPI.
To install this version of fadge
, please use python 3.7 to 3.11
and 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.