Can we create an issue for this? _Originally posted by @michalk8 in https://github.com/theislab/moscot/pull/633#discussion_r1444077840_