Add some type informations to remove non-typechecked parts of the code

Description

This ensures we have type information for cython code and all third parties that provide such type annotations.