Replace `arch_utils` with a reusable NN package containing block-level and low-level neural helpers for explicit internal structure.
Replace
arch_utilswith a reusable NN package containing block-level and low-level neural helpers for explicit internal structure.