Links for flax-vision-models