Update of code for Graph Matching Networks to TF 2 planned?

Hi altogether, we are actively using the approach of Graph Matching Networks in research. Since version 2 of Tensorflow is now the main version and we want to make sure that our code base is future-proof, I was wondering whether an update of the GMN code to TF 2 and the respective version of Sonnet is planned? If not, I am also happy to hear hints on what has to be adjusted.

I hope one of the authors or any other affiliated person can give me a quick answer.

Thank you very much and best regards!