Treasury Secretary Scott Bessent said the U.S. must maintain its lead over China in AI while urging AI companies to better address data center concerns.