SCENIC: A JAX Library for Computer Vision Research and Beyond

Understand

Scenic is an open-source JAX library with a focus on Transformer-based models for computer vision research and beyond.

• The goal of this toolkit is to facilitate rapid experimentation, prototyping, and research of new vision architectures and models.
• Scenic supports a diverse range of vision tasks (e.g., classification, segmentation, detection)and facilitates working on multi-modal problems, along with GPU/TPU support for multi-host, multi-device large-scale training.
• Scenic also offers optimized implementations of state-of-the-art research models spanning a wide range of modalities.