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
466 stars 154 forks source link

[TensorFlow] TensorFlow 2.x - computer vision (CV) CNN model support for inference on Inf1 #421

Open aws-maens opened 2 years ago

aws-maens commented 2 years ago

Add support for computer vision models in tensorflow-neuron 2.x

Othmane796 commented 2 years ago

Any updates on this ? Is it possible to compile Computer vision models built with tensorflow2.x using neuron sdk ? I'm only able to find examples with BERT/HF (here)