Add hetero

2 objectives with this merge request:

1. Add heterogeneous model. This includes the addition of separate submodules to conveniently initiate the components of the heterogeneous model.
2. Factorize all methods to plot performance plots out of the LightningModule into an utils/eval_utils.py. The eval method is still within the LightningModule, but it is configurable with the eval_config.