Plot samples does something reaosnable now... maybe try code that breaks batches of big matrices by page by adding a dummy variable...