charles-r-earp / autograph

A machine learning library for Rust.
Apache License 2.0
315 stars 17 forks source link

candle benches #74

Closed charles-r-earp closed 3 months ago

charles-r-earp commented 3 months ago

Closes #67.

Adds candle to benchmarks with candle and cuda features. Doesn't support bf16, and it appears that tch is a better benchmark.