mirror of
https://github.com/wassname/jaxtyping.git
synced 2026-09-10 12:14:04 +08:00
Patrick Kidger
849b15db3b
Also added support for matching dtypes using regexes, and for nesting e.g. Shaped[Float[Array, "dim1 dim2"], "dim3"].
13 lines
528 B
Markdown
13 lines
528 B
Markdown
If you found this library to be useful in academic work, then please cite: ([arXiv link](https://arxiv.org/abs/2111.00254))
|
|
|
|
```bibtex
|
|
@article{kidger2021equinox,
|
|
author={Patrick Kidger and Cristian Garcia},
|
|
title={{E}quinox: neural networks in {JAX} via callable {P}y{T}rees and filtered transformations},
|
|
year={2021},
|
|
journal={Differentiable Programming workshop at Neural Information Processing Systems 2021}
|
|
}
|
|
```
|
|
|
|
(Also consider starring the project [on GitHub](https://github.com/patrick-kidger/equinox).)
|