Add activation functions to hydra models

This would be nice for hyper-parameter searches and also quick configuration of models from within yaml files.