instead of computing across large batches of pixels
instead of computing across large batches of pixels