@misc{indiciae1b59fb763d84,
  title = {DrJAX: Scalable and Differentiable MapReduce Primitives in JAX},
  author = {Keith Rush and Zachary Charles and Zachary Garrett and Sean Augenstein and Nicole Mitchell},
  year = {2024},
  url = {https://arxiv.org/abs/2403.07128},
  note = {Source identifier: 2403.07128}
}