How to check if scib is using GPU accelerator?

[zhenxingjian]

Description of feature

I would like to ask how can we check if scib is using GPU?

[adamgayoso]

You can check utilization at the system level with nvidia-smi, but also if you run

import jax
jax.devices()

you should see a cuda device.

In general, if jax is using the gpu then this package will as well.