jaxtyping

Community

Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.

Versions

To install this package, run one of the following:

$conda install asford::jaxtyping

Version

1 / 8 versions selected

Downloads (Last 6 months): 0

Summary

Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.

Last Updated

Mar 5, 2024 at 21:48

License

MIT

Supported Platforms

noarch

Home