AI-driven protein folding