py-torchmetrics is a stub port
