Awesome
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