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

support for aten::upsample_nearest3d #908

Open SimonRelu opened 2 weeks ago

SimonRelu commented 2 weeks ago

We're looking into using Inf1 instances on AWS to run our GPU workloads. I was wondering if there are any plans on supporting the aten::upsample_nearest3d (torch) layer since this is the only one that we use that's currently not yet supported.

Another question I had if there are any plans on supporting python 3.11 (and publishing wheels for it) since we recently made the switch.

Thank you for looking into this