This repository provides the official implementation of a CVPR 2022 paper on motion prediction for autonomous vehicles. The model uses a hierarchical transformer architecture to predict future trajectories of multiple agents from vectorized map and agent data. It is trained using PyTorch and PyTorch Lightning on the Argoverse Motion Forecasting dataset, supporting both HiVT-64 and HiVT-128 model variants.