numpy
numba
matplotlib
opencv-python
ipykernel
ipython
notebook
shapely
prime_printer
datasets==3.6.0
scikit-image
img-phy-sim
pytorch-msssim
kornia
mlflow
tensorboard
tqdm
torch
torchvision

[gpu]
torch==2.1.0+cu126
torchvision==0.16.0+cu126
torchaudio==2.1.0+cu126
