A JAX/Equinox re-implementation of the HiVT model for trajectory prediction.
A JAX/Equinox re-implementation of the [HiVT](https://github.com/zhangao0086/HiVT) model for trajectory prediction.