test: removing jax from test-suite#3558

ori-kron-wis merged 3 commits intomainscverse/scvi-tools:mainfrom

Ori-adding-test-suite-nonJaxscverse/scvi-tools:Ori-adding-test-suite-nonJaxCopy head branch name to clipboard