torch
transformers
lightgbm
numpy