谷歌公司設(shè)計了自己的定制芯片，稱為Tensor Processing Unit(TPU)，并將這些芯片應(yīng)用于90%以上的人工智能訓(xùn)練工作。這個過程通過模型對數(shù)據(jù)進行訓(xùn)練，以提高其在類似人類文本響應(yīng)或生成圖像等任務(wù)中的實用性。

　　目前，谷歌TPU已經(jīng)進入第四代。谷歌公司在周二發(fā)布了一篇科學(xué)論文，詳細介紹了如何利用自己開發(fā)的定制光開關(guān)將4000多個芯片連接成一個超級計算機。

　　對于構(gòu)建AI超級計算機的公司來說，改善這些連接已經(jīng)成為競爭的關(guān)鍵點，因為所謂的大規(guī)模語言模型正在不斷增大，已無法在單個芯片上存儲，而這些模型正驅(qū)動著谷歌的Bard或OpenAI的ChatGPT等技術(shù)。

　　這類模型必須分布在成千上萬的芯片之間，然后這些芯片要在數(shù)周或更長時間內(nèi)協(xié)同工作來訓(xùn)練模型。谷歌的PaLM模型，是迄今為止公開披露的最大語言模型，它通過分布在兩臺4000芯片的超級計算機上進行了50天的訓(xùn)練。

　　谷歌表示，其超級計算機可以輕松地動態(tài)重新配置芯片之間的連接，有助于避免故障并進行性能優(yōu)化。

　　“電路交換使得我們能夠輕松避開故障組件，”谷歌高級研究員Norm Jouppi和谷歌杰出工程師David Patterson在一篇關(guān)于該系統(tǒng)的博客文章中寫道。“這種靈活性甚至使我們能夠改變超級計算機互聯(lián)網(wǎng)絡(luò)的拓撲結(jié)構(gòu)，以加速ML（機器學(xué)習(xí)）模型的性能。”

　　雖然谷歌直到現(xiàn)在才公開其超級計算機的詳細信息，但它自2020年以來就已經(jīng)在俄克拉荷馬州梅斯縣的數(shù)據(jù)中心內(nèi)運行。谷歌表示，初創(chuàng)公司Midjourney使用該系統(tǒng)來訓(xùn)練其模型，該模型在輸入幾個詞的文本后可以生成新的圖像。

　　谷歌在論文中表示，與相同規(guī)模的系統(tǒng)相比，它的超級計算機比基于英偉達A100芯片的系統(tǒng)更快1.7倍，更節(jié)能1.9倍。據(jù)了解，A100芯片早于第四代TPU面市。

　　谷歌表示，它沒有將第四代TPU與英偉達當(dāng)前旗艦H100芯片進行比較，因為H100是在谷歌這款芯片應(yīng)用后上市的，并且采用了更新的技術(shù)。

　　谷歌暗示正在研發(fā)一款新的TPU，將與英偉達H100競爭，但沒有提供詳細信息。Jouppi告訴媒體，谷歌有“充足的未來芯片儲備”。