Option for non-vmapped get_local_sample_representation

[justjhong]

Useful if vmap causes OOM on GPU.

[adamgayoso]

LGTM but we can also remove the mc_samples in this linear gaussian setup

[codecov]

Codecov Report

Merging #12 (b6c26a4) into main (72e1a5e) will decrease coverage by 0.29%.
The diff coverage is 90.00%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #12      +/-   ##
==========================================
- Coverage   98.68%   98.38%   -0.30%     
==========================================
  Files           6        6              
  Lines         304      310       +6     
==========================================
+ Hits          300      305       +5     
- Misses          4        5       +1     

Flag	Coverage Δ
unittests	98.38% <90.00%> (-0.30%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
src/scvi_v2/_model.py	98.96% <90.00%> (-1.04%)	⬇️