numpy
torch
scipy
argparse
