aws / aws/amazon-ec2-instance-selector
Filter for selecting Trainium Instances
- Dominant language
- Go
- Stars
- 932
- Forks
- 110
- PR merge metrics
- No merged PRs in 30d
Description
Hi,
There seems to be no filter in the ec2-instance-selector to select the AWS Trainium Instances. It would be of great help if a filter for this can be included. Maybe some filter for `NeuronInfo` as in the describe instance types, similar to `--inference-accelerator` would be fine:
```
$ aws ec2 describe-instance-types --instance-type="trn1.32xlarge"
...
"NeuronInfo": {
"NeuronDevices": [
{
"Count": 16,
"Name": "Trainium",
"CoreInfo": {
"Count": 2,
"Version": 2
},
"MemoryInfo": {
"SizeInMiB": 32768
}
}
],
"TotalNeuronDeviceMemoryInMiB": 524288
},
```
Thanks in advance.
Contributor guide
Assessment
This issue has not been assessed yet.