tqdm
numpy
argparse
neuraloperator
the_well
matplotlib
wandb
