Consider switching from Mypy to Pyright

[eb8680]

Currently we use Mypy for static type checking, but it has several bugs that have required us to disable certain checks or make certain types too imprecise. It also doesn't yet seem to fully support the new generics syntax in Python 3.12, should we wish to adopt that.

Pyright seems to handle these edge cases better, so it might be worth switching.

[jfeser]

Known issues:

• Calling Operations produces error: Expected 0 positional arguments (reportCallIssue)
• Pyro and Torch don't export their symbols in the way pyright expects, so lots of errors like error: "vmap" is not exported from module "torch.func"

[qinqian]

perhaps I can give it a try. I added a branch for this, but not sure how to fix all these errors, so far ignoring them in the configuration.

[eb8680]

Thanks for taking a look. Errors should be addressed unless there's some insurmountable barrier to doing so - there's not much point in switching away from Mypy if we can't strengthen the types in the core library code.

[qinqian]

I see, then I may need some help when fixing some errors that may or may not break the function