Skip to content

Latest commit

 

History

History
14 lines (12 loc) · 301 Bytes

README.md

File metadata and controls

14 lines (12 loc) · 301 Bytes

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