Patched Attention for Nonlinear Dynamics - View it on GitHub