TY  - RPRT
TI  - DrJAX: Scalable and Differentiable MapReduce Primitives in JAX
AU  - Keith Rush
AU  - Zachary Charles
AU  - Zachary Garrett
AU  - Sean Augenstein
AU  - Nicole Mitchell
PY  - 2024
UR  - https://arxiv.org/abs/2403.07128
ID  - 2403.07128
ER  -