refactor: harden imports, add training extras, and rewrite README

first commit