The jaxsim package is widely type hinted, however at the moment it's only useful while developing for IDE suggestions. It would be great if, slowly, we can start having mypy running at least on the jaxsim.api package.
The starting point is not that easy. I get ~200 errors.