dev: Beartype checking.

[daniel-dodd]

As suggested by @vdutor in #21, the integration of beartype checking would be a nice improvement to this library.

[daniel-dodd]

Preferred solution would be option two of https://github.com/google/jaxtyping/blob/main/API.md