Differential geometry for machine learning in JAX.
Pronunciation: Dee Gee JAX (Differential Geometry JAX).
Caution: the package is in early development stage. Expect frequent changes to the API and little documentation.
Install the requirements and the package in the editable mode:
$ pip install -r requirements.txt
$ pre-commit install
$ pip install -e ".[test]"