Putting the AI in IAEA