Topic

Trainium

2 stories

Sections: Tech

Stories

Stage · Medium3 sourcesTech10.20 13:03

Amazon's PyTorch-on-Trainium tool needs one line changed; 22,000 PyTorch tests are skipped

Amazon Web Services released TorchNeuron on Monday, an open-source backend that lets code written in PyTorch run on its Trainium2 and Trainium3 AI chips after a one-line device change.

Stage · Medium2 sourcesTech10.20 01:41

Amazon's TorchNeuron runs PyTorch on Trainium with one change, but 22,000 tests are skipped

Amazon's one-line switch from standard chips to its Trainium AI chips ships with about 22,000 PyTorch tests skipped, according to a PyTorch blog post by Amazon engineers cited by The Next Web.