Use typehints?

Maybe we should adapt typehints in the material? There are a few times where the text "manually" specifies the type of some argument. That would be good places to have typehints instead. Also, it might be helpful for adding numba decorators.