This model was contributed by nielsr. The original code (written in JAX) can be
found here.
Note that we converted the weights from Ross Wightman's timm library, 
who already converted the weights from JAX to PyTorch. Credits go to him!
Usage tips