Flax-Vision Computer Vision Models from PyTorch, Tensorflow now for Flax! Inspired by Ross Wightman's pytorch-image-models.