nvidia-ml-py
scanpy
scvi-tools