numpy
torch
lightning
