optimum-benchmark

A unified multi-backend utility for benchmarking Transformers, Timm, PEFT, Diffusers and Sentence-Transformers with full support of Optimum's hardware optimizations & quantization schemes.

Install from the command line

Learn more about packages


    
  
      
      $ docker pull ghcr.io/huggingface/optimum-benchmark:sha-347e13ca9f7f904f55669603cfb9f0b6c7e8672c-cuda-ort

Recent tagged image versions

sha-347e13ca9f7f904f55669603cfb9f0b6c7e8672c-cuda-ort latest-cuda-ort

Published 7 days ago · Digest
sha256:5a7618e76aacc1405d353da8ed7327b8efeef84fefc738870ee19cd4e8b66a2e

5 Version downloads
sha-347e13ca9f7f904f55669603cfb9f0b6c7e8672c-rocm latest-rocm

Published 7 days ago · Digest
sha256:6bf2bb1c9ccdba2e30f94f646c15b176c7faf00d399de9db104f792851815a41

7 Version downloads
sha-347e13ca9f7f904f55669603cfb9f0b6c7e8672c-cuda latest-cuda

Published 7 days ago · Digest
sha256:04d2b20df2b6a1ebe38aa8d04b4e72e6c321930cc419cfda23abe8dec2fbc008

29 Version downloads
sha-347e13ca9f7f904f55669603cfb9f0b6c7e8672c-cpu latest-cpu

Published 7 days ago · Digest
sha256:edb4cb195094ce37976ca548c1586d88d57dbde65bdd0b8ba419692fd0f663ed

5 Version downloads
sha-dc29eecfd7850e87154a5a92442eb64d237d0070-cuda-ort

Published 21 days ago · Digest
sha256:0934528eb84067a477752a83558817fb777300ed5cf4d34803d5716955a053da

5 Version downloads