Found 12 repositories(showing 12)
patrick-kidger
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
torchmix
A collection of components for transformers 🧩
cowana-ai
Open-source Python library designed to improve engineering practices and transparency in feature engineering.
jonasjuerss
No description available
samsja
LLaMa implementation fork from lit-llama fully typed with jaxtyping
EtaoinWu
enhanced run-time typechecking for `jaxtyping` annotated `equinox` Modules
conda-forge
A conda-smithy repository for jaxtyping.
Yosshi999
mypy plugin for shape inference in the code annotated by jaxtyping
bosque-juan
No description available
redwoodresearch
Type annotations for Jax arrays. Fork of torchtyping.
HarudoBoruzu
JAX + jaxtyping による型安全な Sequential Monte Carlo (SMC) ライブラリ
brentyi
No description available
All 12 repositories loaded