Amazon SageMaker Studio

SageMaker v3 drops the framework estimators, and your training code is the migration

One ModelTrainer and one ModelBuilder replace the SKLearn, PyTorch and XGBoost classes, with source directories synced at job launch. The container is now yours to supply.