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
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support