aws-neuron / aws-neuron-sdk

Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
https://aws.amazon.com/machine-learning/neuron/
Other
420 stars 136 forks source link

Issue on page /frameworks/torch/torch-neuronx/programming-guide/training/pytorch-neuron-programming-guide.html #912

Open yahavb opened 1 week ago

yahavb commented 1 week ago

The note under Automatic Mixed-Precision misleads. It gives the impression that convolution on Neuron is sub-optimal. Can we pls remove it? Specifically, the note is "(NOTE: Although convolution is in the list below, it is currently unsupported by Neuron. See model-architecture-fit)"