Improving the technology behind AI will only increase the return on investment per watt, so you’ll want to spend even more on it than before. This would more than likely increase the energy demands (assuming it doesn’t turn into vaporware).