JAX Autodiff from a Linear Logic Perspective (Extended Version)

Giulia Giusti, Michele Pagani

Autodiff refers to the core of the automatic differentiation systems developed in projects like JAX and Dex. Autodiff has recently been formalised in a linear typed calculus by Rad…