Resolve "torch 2.0.0 doesn't pass with mypy >= 1.9.0"

Summary

(Summarize the merge request concisely.)

Explain your context

(Explain what you did.)

Please list breaking changes, new features or bugfixes

• Breaking changes:

• New features:

• Bugfixes:

Closes #269 (closed)