Google представил собственный процессор Axion для ИИ

Google создает собственный процессор на базе Arm для поддержки работы ИИ в центрах обработки данных и представляет более мощную версию своих ИИ-чипов Tensor Processing Units (TPU). Новый процессор Google на базе Arm, получивший название Axion, будет использоваться для поддержки рабочих нагрузок Google в области искусственного интеллекта, прежде чем он будет распространен среди бизнес-клиентов Google Cloud «позднее в этом году».

Чипы Axion уже используются в рекламной сети YouTube, движке Google Earth Engine и других сервисах Google. «Мы упрощаем заказчикам перенос существующих рабочих нагрузок на Arm», — говорит Марк Ломайер, вице-президент и генеральный менеджер Google Cloud по вычислениям и инфраструктуре машинного обучения. «Axion построен на открытых основах, поэтому клиенты, использующие Arm в любом месте, смогут легко начать работать с ним, не перестраивая архитектуру и не переписывая свои приложения».

По словам Google, клиенты смогут использовать процессор Axion в таких облачных сервисах, как Google Compute Engine, Google Kubernetes Engine, Dataproc, Dataflow, Cloud Batch и других. По сообщению Reuters, процессор Axion на базе Arm также будет на 30% производительнее «чипов Arm общего назначения» и на 50% — существующих процессоров Intel.

Google также обновляет свои чипы TPU AI, которые используются в качестве альтернативы графическим процессорам Nvidia для задач ускорения ИИ. «TPU v5p — это ускоритель нового поколения, созданный специально для обучения самых больших и требовательных генеративных моделей ИИ», — говорит Лохмайер. Одна капсула TPU v5p содержит 8 960 чипов, что более чем в два раза превышает количество чипов в капсуле TPU v4.

Анонс Google процессора на базе Arm состоялся через несколько месяцев после того, как Microsoft представила собственные процессорый, предназначенные для облачной инфраструктуры. Microsoft создала свой собственный чип для обучения больших языковых моделей ИИ и собственный процессор на базе Arm для облачных и ИИ-нагрузок. Amazon также уже несколько лет предлагает серверы на базе  собственных Arm процессоров.

Google не будет продавать эти чипы клиентам, вместо этого они будут доступны для облачных сервисов, которые компании могут арендовать и использовать. «Стать великой компанией по производству аппаратного обеспечения — это совсем другое, чем стать великой облачной компанией или великим организатором мировой информации», — сказал Амин Вахдат, руководитель подразделения Google по производству чипов.

Google, как до нее Microsoft и Amazon, теперь может уменьшить свою зависимость от таких партнеров, как Intel и Nvidia, а также конкурировать с ними в области создания собственных чипов для обеспечения работы искусственного интеллекта и облачных вычислений.