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: CoRR (2021)