TY  - RPRT
TI  - Equinox: neural networks in JAX via callable PyTrees and filtered transformations
AU  - Patrick Kidger
AU  - Cristian Garcia
PY  - 2021
UR  - https://arxiv.org/abs/2111.00254
ID  - 2111.00254
ER  -