I've been working on this package last year, it will be presented at ACL 2023: https://github.com/IBM/zshot