Amazon SageMaker Neo now supports compilation of PyTorch and TensorFlow models for Inferentia 2 and Trainium 1 instances. You can quickly get started through the console simply by selecting ml_inf2 or ml_trn1 as the Target Device.
Sort:
Amazon SageMaker Neo now supports compilation of PyTorch and TensorFlow models for Inferentia 2 and Trainium 1 instances. You can quickly get started through the console simply by selecting ml_inf2 or ml_trn1 as the Target Device.
Sort: