Refactor hacked Jacobian into a BatchedDot. Huge speedup.
Refactor hacked Jacobian into a BatchedDot. Huge speedup.

nyu-mll/spinn,stanfordnlp/spinn,stanfordnlp/spinn,nyu-mll/spinn,nyu-mll/spinn,stanfordnlp/spinn