Make loading faster:

use multiprocessing
trying to figure out why GPU copies are so slow.