/benchmarks

a series of benchmarks for shumai

Primary LanguageJavaScript

shumai_benchmarks

a series of benchmarks for shumai

To install things:

npm install @tensorflow/tfjs-node # or @tensorflow/tfjs-node-gpu
bun install @shumai/shumai
pip install torch # you may need to install tensorflow to get GPU support on tfjs

To run them:

python run.py show_cmd