---
license: cc-by-4.0
---

CausalTraj models are trained for joint multi-agent trajectory forecasting in team sports.

This repo contains 2 variants of of model weights: CausalTraj (Causal PointNet) and CausalTraj (Mamba2), both trained on the NBA SportVU dataset.

You can find the corresponding code base to load the models here: https://github.com/wezteoh/causaltraj