Could AI help us create imagination machines?