AI infrastructure for large language models