Scenic: A Jax Library for Computer Vision Research and Beyond - google-research/scenic