Check If Your Hugging Face Model Runs on AWS Trainium
What is Does Your Model Run on AWS Trainium? Look It Up?
The new TorchNeuronx Space lists which Hugging Face models can run unmodified on AWS Trainium chips. It shows sharding, wall-time (including compilation), peak memory and provides a ready-to-run script.
Who it is for
PyTorch developers targeting Trainium can instantly verify compatibility and copy the script to benchmark or deploy their model.
How to try it
Open https://huggingface.co/spaces/huggingface/torch-neuronx-model-support