Amazon Webes szolgáltatások (AWS) újabb újítást kap a nemrég bemutatott, Mac minin alapuló macOS példányok mellett. Ez a következőkről szól Trainiumaz Amazon felhőalapú számítástechnikai részlegének chipje, amelyet kifejezetten neurális hálózatok képzésére fejlesztettek ki. Az Inferentia ASIC-khez és a Graviton2 CPU-khoz hasonlóan a Trainium is saját fejlesztés.
Az ügyfelek 2021 második felétől használhatják a Trainiumot. Az Amazon szerint a fejlesztés során a hangsúlyt az árhoz képest nagyon jó teljesítményre helyezték. Az Amazon szerint a példányoknak az Intel Habana Gaudiával rendelkező példányok alatt kell lenniük, amelyek viszont már 40 százalékkal jobb ár-teljesítmény aránnyal rendelkeznek, mint a p4d.24xlarge konfiguráció nyolc Nvidia A100-as processzorral.
Műszaki adatok ismeretlenek
Az Amazon még nem tett közzé részleteket a Trainium chip technológiájáról és teljesítményéről. Annyit tudni, hogy az AWS példányok a vállalat összes gépi tanulási példánya közül a legnagyobb teraflopos teljesítményűek lesznek.
Az is ismert, hogy a Felhő A Pytorch, MXNet és Tensorflow szabványos keretrendszerek támogatása. A Neuron nevű szoftverfejlesztő készlet (SDK) szintén ismert már az Inferentia ASIC-ekből.
A Trainium elérhető a Sagemaker szolgáltatásban és az EC2 példányokban (Elastic Compute Cloud). Az AWS-en gyakran használt tanulási algoritmusok és keretrendszerek állítólag előre telepítve vannak.