Skip to main content

Supported ML frameworks

Layer supports the following machine learning frameworks:

  • CatBoost
  • Keras
  • LightGBM
  • PyTorch
  • scikit-learn (without XGBoost)
  • TensorFlow & TensorFlow Tokenizers
  • Hugging Face Transformers & Tokenizers
  • XGBoost version 0.9.0 or later
  • YOLOv5