Hi, How easy/difficult it can be to to use the framework to train and use other models e.g. Bert or like? Best Regards,