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