scvi: NaNs on apple M2 silicon

[sjfleming]

Not that it's a top priority to understand the idiosyncrasies of training on M2 silicon, but there areNaN values that pop up very quickly when training scVI on M2. These do not appear when training on CPU or GPU.