The AI that could help make fusion power a reality