- Add docstrings, at least to the API functions. - Use sphinx to generate documentation pages like Numpy and Jax do.