# RNNs perform task computations by warping neural representations

The code presented here can be used to make figures 2, 4 and 6 of the paper. Figures 1 and 5 are schematics.

## Requirements

To install the riemannian_dynamics package along with the requirments you can

```pip install path_to_directory/riemannian_dynamics```

## Running the code

* `/projects/riemannian_dynamics` contain general files to train and plot models
* `/projects/static_network` contains the network of figure 2
* `/projects/dynamic_networks.py` contains the RNNs of figure 3 and 5

All figures can be obtained by running the notebooks within these folders. The computation of the metric can take several minutes (but usually under an hour).

## Pre-trained models

Pre-trained models are provided in the above folders.
