Build a batched version of the MLP