StateTransformer
Multi‑agent trajectory prediction model for autonomous driving.
StateTransformer is a transformer‑based model designed for trajectory prediction in self‑driving scenarios. It integrates rasterized map data, agent context, and temporal dynamics to generate accurate future trajectories.
Not supported
This model is currently not supported on any Mobile chipset.
To see performance metrics for this model on other chipsets, click the button below.
View for other chipsetsTechnical Details
Input resolution:1x224x224x58, 1x224x224x58, 1x4x7
Model checkpoint:pretrained-mixtral-small
Model size (float):348 MB
Number of parameters:90.7M
Applicable Scenarios
- Self driving cars
License
Model:MIT
Supported Mobile Devices
- Samsung Galaxy S21
- Samsung Galaxy S21 Ultra
- Samsung Galaxy S22 5G
- Samsung Galaxy S22 Ultra 5G
- Samsung Galaxy S22+ 5G
- Samsung Galaxy S23
- Samsung Galaxy S23 Ultra
- Samsung Galaxy S23+
- Samsung Galaxy S24
- Samsung Galaxy S24 Ultra
- Samsung Galaxy S24+
- Samsung Galaxy S25
- Samsung Galaxy S25 Ultra
- Samsung Galaxy S25+
- Samsung Galaxy S26
- Samsung Galaxy S26 Ultra
- Samsung Galaxy S26+
- Samsung Galaxy Tab S8
- Snapdragon 8 Elite Gen 5 QRD
- Snapdragon 8 Elite QRD
- Xiaomi 12
Supported Mobile Chipsets
- Snapdragon® 8 Elite Mobile
- Snapdragon® 8 Elite Gen 5 Mobile
- Snapdragon® 8 Gen 1 Mobile
- Snapdragon® 8 Gen 2 Mobile
- Snapdragon® 8 Gen 3 Mobile
- Snapdragon® 888 Mobile
Looking for more? See models created by industry leaders.
Discover Model Makers