Multi GPU inference support

[alehuerta1]

Is there support for multi-GPU inference in SCVI? Say I wanted to add distributed_sampler=True to the arguments in make_data_loader in get_latent_representation(). Is this possible? If not, how would you go about implementing this functionality?

[ori-kron-wis]

Hey,
There isnt right now, and its a very simple solution to add it so its a good idea.
You will have to add a **data_loader_kwargs parameter to the get_latent_representation, or any other downstream function that will use it in the make_data_loader generic function calling.

See: #3443

[canergen]

Is the order still correct of cells? Or do you have to reorder your cells to align with adata.obs_names (this step is not trivial as we don't track indices).

[ori-kron-wis]

@alehuerta1 can we close?