Would be nice to have everything statically typed. Note that all params and return values are all implicitly typed from the docstrings (and hence the docs), so for those that aren't typed, it would be just a matter of corroborating docstrings and type hints.

Also, we should add mypy to the CI to check this.

This will complement the ruff checker added in #501