AWS releases TorchNeuron to run PyTorch AI applications on Trainium chips
Amazon Web Services released TorchNeuron, which it says lets developers move PyTorch AI applications to its Trainium chips with a configuration change.
Amazon Web Services announced TorchNeuron, a tool designed to let developers run PyTorch-based AI applications on AWS Trainium chips without rewriting their code.
AWS said it also plans to contribute changes intended to make PyTorch less tied to particular hardware.