@article{aresearchframeworkforwriting,
  title = {A research framework for writing differentiable PDE discretizations in JAX},
  author = {Antonio Stanziola and Simon R. Arridge and Ben T. Cox and Bradley E. Treeby},
  year = {2021},
  eprint = {2111.05218},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2111.05218v1},
}