Bringing AI to the next generation of fusion energy