← all repositories

google-research/scenic

A JAX-based library for training and evaluating attention-based vision models at scale, supporting classification, segmentation, detection, and multimodal tasks.

3.8k stars Python Computer VisionML Frameworks
scenic
Velocity · 7d
+2.1
★ / day
Trend
steady
star history

Scenic provides shared lightweight libraries and project templates for training large-scale vision models using JAX and Flax. It implements state-of-the-art models including vision transformers (ViViT), video models, and multimodal architectures. The library includes optimized training loops, losses, metrics, input pipelines for vision datasets, and baseline implementations for common computer vision tasks.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.