Getting to the point: index sets and parallelism-preserving autodiff for pointful array programming.
Adam PaszkeDaniel D. JohnsonDavid DuvenaudDimitrios VytiniotisAlexey RadulMatthew J. JohnsonJonathan Ragan-KelleyDougal MaclaurinPublished in: Proc. ACM Program. Lang. (2021)