@misc{indiciaed210ee54b1a7,
  title = {Equinox: neural networks in JAX via callable PyTrees and filtered transformations},
  author = {Patrick Kidger and Cristian Garcia},
  year = {2021},
  url = {https://arxiv.org/abs/2111.00254},
  note = {Source identifier: 2111.00254}
}