Thank you! MLEM already supports Scikit-learn, PyTorch, XGBoost, LightGBM and CatBoost now. We're going to add fastai, TensorFlow and Huggingface very soon, and then there are many others we'd like to support! Feel free to post an issue in the GH repo for the framework of your choice)