Adding type stubs

For your reference: https://github.com/python/typing/discussions/1858
I'm not sure if this project is accepting contributions, so I'm thinking of adding the type stubs outside of it.