Fast Diagonal Convolutions (in JAX)

Why Convolutions?