Pytorch Implementation of Dycheck

[guanjunwu]

Hi,
Thanks for your great work!
I found it difficult to install the full codebase because of jaxlib implementation. Is there any plan to release pytorch implementation of dycheck dataset?
Meanwhile, can you provide a script to guide us to create a custom dataset like your format?
Best,
guanjun