AI 行业资讯 - PyTorch

TorchNeuron公开测试版将发布，Trainium原生运行PyTorch模型

AWS宣布TorchNeuron将于10月20日作为公开测试版发布。该PyTorch插件为Trainium提供新后端，支持eager模式、torch.compile及FSDP等原生分布式API。Hugging Face的Transformers、Diffusers等库现已原生支持，用户可直接使用，无需再通过Optimum Neuron。