Depending on your model, you may spend a lot of time trying to get it to work with Trainium