lightgbm
pytorch-metric-learning
torch
transformers
PyTDC
gradio
numpy