RecodeX 重构消息,AWS发布TorchNeuron公开测试版,为Trainium2和Trainium3芯片提供PyTorch后端支持。该后端支持即时执行(eager execution)、编译以及分布式API。不过,编译后的模型需要静态形状,且部分API目前仍不受支持。