Rethink some code annotations

[adam-urbanczyk]

In order to make CQ more friendly for codbases that use strict type checking (i.e. not for typical modeling):

1. Consider using overloads instead of mypy plugins (if possible)
2. Assembly[...] overloads based on tags or string subtypes