Jaxtyping - Githubissues

danielward27 / flowjax

https://danielward27.github.io/flowjax/

MIT License

82 stars 10 forks source link

Closed danielward27 closed 4 months ago

danielward27 commented 4 months ago

Converts to making use of jaxtyping. The tests are now run using beartype so that type annotations are enforced/checked to be correct during testing.