First release (and paper)

[ismael-mendoza]

Tasks related to releasing a first version of the code along with a companion paper

• Clean up code #88
• demo1.py #90
• demo2.py #91
• demo1-jax.py #92 (optional, but would be cool)
• Setup documentation system to keep track of implemented API #17
• update README #89
• Release and push to pypi, conda-forge