🧠

AWS Neuron / Trainium

2 test cases

NeuronX Distributed and Optimum Neuron for training on AWS Trainium chips with optimized compilers, tensor parallelism, and pipeline parallelism.

🧠

NeuronX Distributed

Distributed training on AWS Trainium with NeuronX

NeuronXTrainiumInferentiaCustom Silicon
🧠

Optimum Neuron

HuggingFace Optimum for training and inference on AWS Trainium and Inferentia

Optimum NeuronHuggingFaceTrainiumInferentia