🌍Optimum Neuron

🌍 Optimum Neuron

🌍 Optimum Neuron is the interface between the 🌍 Transformers library and AWS Accelerators including AWS Trainium and AWS Inferentia. It provides a set of tools enabling easy model loading, training and inference on single- and multi-Accelerator settings for different downstream tasks. The list of officially validated models and tasks is available here.