AWS, Trainium yapay zekâ çipleri için yerel bir PyTorch arka ucu olan TorchNeuron’ın genel beta sürümünü duyurdu. Açık kaynaklı kütüphane, geliştiricilerin .to('neuron') ekleyerek PyTorch kodlarını Trainium’da çalıştırmasına olanak tanıyor. Eager execution ve torch.compile gibi standart PyTorch özellikleriyle uyumlu. TorchNeuron, GitHub deposu üzerinden EC2 Trainium2 ve Trainium3 bulut sunucularında kullanılabiliyor.
Neden önemli
Araştırmacılar artık kodlarını yeniden yazmadan PyTorch modellerini AWS’nin yüksek performanslı Trainium çiplerinde eğitebiliyor; bu da maliyeti ve gereken çabayı azaltıyor.
