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 Automotive chipset.

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

View for other chipsets

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 Automotive Devices

  • SA7255P ADP
  • SA8255P ADP
  • SA8295P ADP
  • SA8650P ADP
  • SA8775P ADP

Supported Automotive Chipsets

  • Qualcomm® SA7255P
  • Qualcomm® SA8295P
  • Qualcomm® SA8775P

Looking for more? See models created by industry leaders.

Discover Model Makers