Basic JaxPEAKVI class

[RK900]

Partially addresses #1578

[codecov[bot]]

Codecov Report

Merging #1608 (934ff81) into master (b9f5b84) will increase coverage by 0.12%. The diff coverage is 97.87%.

@@            Coverage Diff             @@
##           master    #1608      +/-   ##
==========================================
+ Coverage   91.39%   91.51%   +0.12%     
==========================================
  Files         116      118       +2     
  Lines        9190     9371     +181     
==========================================
+ Hits         8399     8576     +177     
- Misses        791      795       +4     

Impacted Files	Coverage Δ
scvi/nn/__init__.py	100.00% <ø> (ø)
scvi/model/_jaxpeakvi.py	95.91% <95.91%> (ø)
scvi/module/_jaxpeakvae.py	98.43% <98.43%> (ø)
scvi/model/__init__.py	100.00% <100.00%> (ø)
scvi/module/__init__.py	100.00% <100.00%> (ø)
scvi/module/_jaxvae.py	100.00% <100.00%> (ø)
scvi/nn/_base_components.py	94.48% <100.00%> (+0.16%)	:arrow_up:

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[RK900]

Latent space from JPVI, matching the one from the PeakVI tutorial