Update mem/compute scripts interaction

Move all the model setup in a different script. Add per architecture
variables (for example, feed forward matrices size), since most of the
architectures today vary in one way or another.

This makes it easier to change values around and get more meaningful
results, and also enables users to more easily add new models.