Про PyTorch пишуть:

Amazon запускає свої чіпи AWS Trainium для всіх розробників PyTorch без переписування коду

TorchNeuron дозволяє розробникам запускати існуючий AI-код на чіпах AWS Trainium практично без змін, спрощуючи інтеграці...