AWS از عرضه عمومی نسخه بتای TorchNeuron خبر داد؛ بک‌اند بومی PyTorch برای تراشه‌های Trainium این شرکت. این کتابخانه متن‌باز به توسعه‌دهندگان اجازه می‌دهد تنها با افزودن .to('neuron')، کد PyTorch را روی Trainium اجرا کنند. TorchNeuron با قابلیت‌های استاندارد PyTorch، از جمله اجرای eager و torch.compile، سازگار است. این ابزار از طریق مخزن GitHub برای نمونه‌های EC2 مبتنی بر Trainium2 و Trainium3 در دسترس است.
چرا مهم است
پژوهشگران اکنون می‌توانند بدون بازنویسی کد، مدل‌های PyTorch را روی تراشه‌های پرقدرت Trainium آمازون آموزش دهند؛ این کار هزینه و زحمت را کاهش می‌دهد.
