You don't need AI, just use stronger magnets[1]. ITER was conceived before high temperature superconducting magnets were discovered, so it has more trouble maintaining plasma stability.

1: https://www.youtube.com/watch?v=KkpqA8yG9T4&t=39m0s

MIT-affiliated company building a reactor using these magnets: https://cfs.energy/technology