@sbmaruf  This is a great idea. I am really interested in it and want to join the project. I have some experience fine-tuning the transformers and I am interested in learning the new framework to speed up the training process.