feat: Support for MLX in scvi model

[ori-kron-wis]

close #3268

[codecov]

Codecov Report

❌ Patch coverage is 0.26316% with 379 lines in your changes missing coverage. Please review.
✅ Project coverage is 73.42%. Comparing base (370c642) to head (c4c7482).

Files with missing lines	Patch %	Lines
src/scvi/module/_mlxvae.py	0.00%	146 Missing ⚠️
src/scvi/train/_trainingplans.py	1.06%	93 Missing ⚠️
src/scvi/model/base/_mlxmixin.py	0.00%	68 Missing ⚠️
src/scvi/model/_mlxscvi.py	0.00%	53 Missing ⚠️
src/scvi/model/__init__.py	0.00%	5 Missing ⚠️
src/scvi/model/base/__init__.py	0.00%	5 Missing ⚠️
src/scvi/module/__init__.py	0.00%	5 Missing ⚠️
src/scvi/train/__init__.py	0.00%	4 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3598      +/-   ##
==========================================
- Coverage   74.70%   73.42%   -1.28%     
==========================================
  Files         225      228       +3     
  Lines       21631    22011     +380     
==========================================
+ Hits        16159    16162       +3     
- Misses       5472     5849     +377     

Files with missing lines	Coverage Δ
src/scvi/train/__init__.py	76.19% <0.00%> (-17.93%)	⬇️
src/scvi/model/__init__.py	77.77% <0.00%> (-17.68%)	⬇️
src/scvi/model/base/__init__.py	75.00% <0.00%> (-19.74%)	⬇️
src/scvi/module/__init__.py	76.92% <0.00%> (-18.32%)	⬇️
src/scvi/model/_mlxscvi.py	0.00% <0.00%> (ø)
src/scvi/model/base/_mlxmixin.py	0.00% <0.00%> (ø)
src/scvi/train/_trainingplans.py	73.56% <1.06%> (-11.33%)	⬇️
src/scvi/module/_mlxvae.py	0.00% <0.00%> (ø)

... and 1 file with indirect coverage changes

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[ori-kron-wis]

umaps (based on api_overview data):