Following the instructions provided in this colab notebook, I downloaded a pre-trained TF model. Now, I want to convert it to pytorch. I tried several options for the conversion, including ONNX, but I am unable to do so. When I spoke to the authors of the repo, they said they trained the model in Jax and converted it to TensorFlow for the release.