image-reward
torchmetrics
