README.md

---
license: mit
language:
- en
---
All model code and instructions can be found here: https://github.com/anayebi/mental-sim.

This repo contains pretrained PyTorch models that are optimized to predict the future state of their environment.
As these models are stimulus-computable, they can be ``dropped in'' on any new video for further neural and behavioral comparisons of your design.

This repository is based on our paper:

**Aran Nayebi, Rishi Rajalingham, Mehrdad Jazayeri, Guangyu Robert Yang**

["Neural foundations of mental simulation: future prediction of latent representations on dynamic scenes"](https://arxiv.org/abs/2305.11772)

*37th Conference on Neural Information Processing Systems (NeurIPS 2023). Selected for spotlight.*

Here's a [video recording](https://youtu.be/9h_3bHVDMhA?t=1756) that explains our work a bit.