scvi-tools icon indicating copy to clipboard operation
scvi-tools copied to clipboard

Published 20 hours ago • verified publisher icon scverse

Implementation of PeakVI in Jax

Open justjhong opened this issue 3 years ago • 0 comments

Reimplementation of PeakVI in Jax. This is meant to both see if we can get a similar speed up in PeakVI, as well as understanding the limitations of the JaxModuleWrapper abstraction.

justjhong avatar Jun 23 '22 18:06 justjhong