GitHub repository: sablokgaurav/scenic
Description: Scenic: A Jax Library for Computer Vision Research and Beyond