Addition of Pretrained models: ResNet

[MrinalTyagi]

Hi team. As the flax library is great for nn in JAX, I wanted to know your thoughts on the addition of deep learning pre-trained models in the library. In case the following task is in the pipeline, I would surely love to take this up.

Resnet Paper: https://arxiv.org/pdf/1512.03385.pdf