TensorTonic
Problems

JVP and VJP for a 2-Layer Network

Compute Jacobian-vector and vector-Jacobian products through a two-layer ReLU network without forming the full Jacobian.