Move training/runtime logic into a dedicated package for cleaner separation between architectures and training flow.
Move training/runtime logic into a dedicated package for cleaner separation between architectures and training flow.