AutoMate: Specialist and Generalist Assembly Policies over Diverse Geometries