Regarding mentioning a TF2 implementation

[sayakpaul]

Hi @PForet. Thank you for open-sourcing SAM.

I have been dabbling with SAM for the past few days and I am pleased to share a minimal TF2 implementation of SAM. The repo includes a Colab Notebook demoing SAM on CIFAR10 with TPUv2-8 hardware.

Is it possible to mention this from the README? It might help the community to quickly experiment with SAM and expand on it.