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-6e6b10363f3ac65926881f2c6a6113b6cefc06cd-cuda

Recent tagged image versions

sha-6e6b10363f3ac65926881f2c6a6113b6cefc06cd-cuda latest-cuda

Published about 8 hours ago · Digest
sha256:6e65a4d78e654a18cf05f8b2bf81988d92fd657a03eed0856dcaa2fe52658d79

1 Version downloads
sha-6e6b10363f3ac65926881f2c6a6113b6cefc06cd-rocm latest-rocm

Published about 8 hours ago · Digest
sha256:7e9f42220d957fcb2875a88aecb124284f18c55dcda47b7c371c1663e0f0ebfb

1 Version downloads
sha-6e6b10363f3ac65926881f2c6a6113b6cefc06cd-cuda-ort latest-cuda-ort

Published about 8 hours ago · Digest
sha256:150f90ea63bdfd5c5e1120eeed45304fd0147c517db4a68e6a3e8f46a080dacb

1 Version downloads
sha-6e6b10363f3ac65926881f2c6a6113b6cefc06cd-cpu latest-cpu

Published about 9 hours ago · Digest
sha256:4962e6dac29e7700641a1ed49191d028b81b5f551958c3e52c69f1464c8ee9e7

1 Version downloads
sha-9104793fa9ba932870d9ceb029cd628a1388b11e-cuda

Published 1 day ago · Digest
sha256:a04766028fe277b4f0dc0fd3f1cca395eb81fa3a89f4158101e1ea67acbf14ca

13 Version downloads