Hello all, Sorry if this is a silly question, but I'm working on PR to add trainium (

trn1

) integration. But looking through these docs (https://docs.aws.amazon.com/AmazonECS/latest/developerguide/ecs-inference.html), it looks like trainium uses the same device paths as inferentia. So given that, there is no difference in the tasks definitions between inferentia and trainium. So is it fair to say that the already existing flag for inferentia would then likely work for the trainium chip? Will test it out now to see