Unified version of 2nd MLP architecture