v0.5.0

Potentially breaking changes:

• Significant internal refactoring, notably splitting stax into multiple sub-modules, and moving implementations into an _src folder. This could break your code if you use internal function like nt.utils.typing, nt.utils.utils, nt.utils.Kernel etc. (public API will remain unchanged). This should be easily fixed by updating the imports, e.g. nt.utils -> nt._src.utils.

New features:

• Improve documentation with (sub-)sections and better type annotations.
• Improve type annotations via typing protocols
• Run pytype in GitHub Actions