Simple PyTorch matmul benchmark CLI tool - View it on GitHub