Latest jax 0.7.0 (from July 22nd) has breaking change for current scvi (version 1.3.2) #105
Description
Report
Can be fixed by downgrading jax to 0.6.2, e.g. via:
pip install "jax==0.6.2"
(restarting environment might be needed)
Version information
No response