Backport PR #3558 on branch 1.4.x (test: removing jax from test-suite)#3559

ori-kron-wis merged 1 commit intoscverse:1.4.xscverse/scvi-tools:1.4.xfrom

meeseeksmachine:auto-backport-of-pr-3558-on-1.4.xmeeseeksmachine/scvi-tools:auto-backport-of-pr-3558-on-1.4.xCopy head branch name to clipboard