Matthew J. Johnson

Parallelism-preserving automatic differentiation for second-order array languages

Adam Paszke, Dougal Maclaurin, Roy Frostig, 2021, FHPNC@ICFP.

You Only Linearize Once: Tangents Transpose to Gradients

Matthew J. Johnson, Roy Frostig, Adam Paszke, 2022, 2204.10923.

Dex: array programming with typed indices

Matthew J. Johnson, Alexey Radul, Dougal Maclaurin, 2019 .

Getting to the point: index sets and parallelism-preserving autodiff for pointful array programming

Daniel D. Johnson, Matthew J. Johnson, D. Duvenaud, 2021, Proc. ACM Program. Lang..

You Only Linearize Once: Tangents Transpose to Gradients

Matthew J. Johnson, Roy Frostig, Adam Paszke, 2022, Proc. ACM Program. Lang..

Getting to the point: index sets and parallelism-preserving autodiff for pointful array programming

Daniel D. Johnson, Matthew J. Johnson, D. Duvenaud, 2021, Proc. ACM Program. Lang..