Is there a way to use it with Pytorch?

[vshlroot]

Hi

I am interested in using RDDLGym in PyTorch. It seems this codebase is tightly coupled with TF 1. Where do you recommend I start in order to build a framework independent RDDLGym such that it can be used like any other RDDLGym environment and analogous to any DL library?

Thanks in advance!