Add type hints throughout the repo

[johnbcoughlin]

• Add type hints to all functions
• Use the new typechecker ty from astral.sh to validate correctness of newly added type hints after every modification