* enable mypy in setup.cfg

* add some more type annotations

* add py.typed

* add mypy to ci

* fix some type annotations