Optimum-neuron

Latest version: v0.1.0

Safety actively analyzes 715174 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 6 of 6

0.0.1

The following architectures can be trained on [AWS Trainium instances](https://aws.amazon.com/fr/ec2/instance-types/trn1/) (trn1.2xlarge and trn1.32xlarge) :

- ALBERT
- BERT
- DistilBERT
- RoBERTa
- XLM-RoBERTa
- CamemBERT
- Electra
- GPT-2
- GPT-Neo
- MarianMT
- T5
- BART
- ViT

Training examples for many tasks are provided [here](https://github.com/huggingface/optimum-neuron/tree/v0.0.1-release/examples).

Page 6 of 6

© 2025 Safety CLI Cybersecurity Inc. All Rights Reserved.