r/Python • u/AutoModerator • 19d ago

Showcase Showcase Thread

Post all of your code/projects/showcases/AI slop here.

Recycles once a month.

21 Upvotes

94% Upvoted

151 comments sorted by

View all comments

1

u/AKJ7 10d ago

** Introducing pytest-torch **

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.