With a considerable amount of help from Willi Gierke, we were able to port the original
.npz weights of new BiT-ResNet models  to TensorFlow and Keras.
Colab Notebook showing the porting and all the other details are available here . The converted models are available on TF-Hub as well:
These models are a part of the recent work done by Google Brain on Knowledge Distillation .
 BiT-jax2tf: https://github.com/sayakpaul/BiT-jax2tf