Test that compare CPU/GPU speeds needed

[ilibarra]

• comparison notebook using CPU vs GPUs and increasing number of cells required. Focus on memory/time and locate in docs/notebooks
• and visualization of batch effect sizes in overall training time.