jax

0786da8f - Removed unnecessary mypy exclusions from pyproject.toml

Removed unnecessary mypy exclusions from pyproject.toml * 2/3 files type check just fine now * the remaining one could be handled via a file-level directive