torch
torchvision
numpy
pandas
opencv-python
pillow
argparse
pyyaml
scikit-learn
wandb
uuid
pytorch-lightning
pytorch-lightning-bolts
ray[tune]
boto3
botocore
sagemaker
python-dotenv
