Add functions for extracting gene embeddings, move state_embs_dict outside isp, fix bugs in isp

Add option to output embs as tensor

Add setup.py