Qualcomm® AI HubAI Hub

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.

Model RepositoryHugging Face

Technical Details

Model checkpoint:pretrained-mixtral-small
Input resolution:1x224x224x58, 1x224x224x58, 1x4x7
Number of parameters:90.7M
Model size (float):348 MB

Applicable Scenarios

  • Self driving cars

Licenses

Deployable Model:AI-HUB-MODELS-LICENSE

Supported Devices

  • QCS8275 (Proxy)
  • QCS8450 (Proxy)
  • QCS8550 (Proxy)
  • QCS9075 (Proxy)
  • Samsung Galaxy S21
  • Samsung Galaxy S21 Ultra
  • Samsung Galaxy S21+
  • 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 Tab S8
  • Snapdragon 8 Elite Gen 5 QRD
  • Snapdragon X Elite CRD
  • Snapdragon X Plus 8-Core CRD
  • Xiaomi 12
  • Xiaomi 12 Pro
  • XR2 Gen 2 (Proxy)

Supported Chipsets

  • Qualcomm® QCS8275 (Proxy)
  • Qualcomm® QCS8550 (Proxy)
  • Qualcomm® QCS9075 (Proxy)
  • Snapdragon® 8 Elite For Galaxy Mobile
  • Snapdragon® 8 Elite Gen5 Mobile
  • Snapdragon® 8 Gen 1 Mobile
  • Snapdragon® 8 Gen 2 Mobile
  • Snapdragon® 8 Gen 3 Mobile
  • Snapdragon® 888 Mobile
  • Snapdragon® X Elite
  • Snapdragon® X Plus 8-Core

Looking for more? See models created by industry leaders.

Discover Model Makers