Ecosyste.ms: Packages

An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

conda-forge.org : jaxtyping

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

Registry - Source - JSON
purl: pkg:conda/jaxtyping
Keywords: jax, python-typing, typing
License: MIT
Latest release: over 1 year ago
First release: over 1 year ago
Dependent packages: 1
Stars: 438 on GitHub
Forks: 18 on GitHub
See more repository details: repos.ecosyste.ms
Last synced: 6 days ago

equinox 0.9.2 💰

Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.
16 versions - Latest release: over 1 year ago - 1 dependent package - 1,049 stars on GitHub