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.

Download Model
Not supported

This model is currently not supported on any All Models chipset.

To see performance metrics for this model on other chipsets, click the button below.

View for other chipsets
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

License

Model:MIT

Supported Devices

  • Dragonwing IQ-9075 EVK
  • Dragonwing Q-8750
  • QCS8275 (Proxy)
  • QCS8450 (Proxy)
  • QCS8550 (Proxy)
  • 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 Tab S8
  • Snapdragon 8 Elite Gen 5 QRD
  • Xiaomi 12
  • Xiaomi 12 Pro
  • XR2 Gen 2 (Proxy)

Supported Chipsets

  • Qualcomm® QCS8275 (Proxy)
  • Qualcomm® QCS8550 (Proxy)
  • Qualcomm® QCS9075
  • 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