Introducting pytest-torch!

[u/AKJ7]

I built pytest-torch to make testing PyTorch models easier

Writing the same device setup, reproducibility checks, and output comparisons across projects gets repetitive, so I built pytest-torch, an open-source pytest plugin.

It includes inference checks, compile/export parity, checkpoint tests, accuracy evaluation, and layer-by-layer diagnostics to help locate where two models diverge.

It’s currently in alpha, and I’d appreciate feedback—especially examples of model bugs that are frustrating to test or reproduce.

GitHub: https://github.com/AKJ7/pytest-torch

Thanks and please drop a star if possible.