AI infrastructure for data centers