node.js >= 12.x
python >= 3.8
# install dependencies
npm run install:all
# build
npm run build
# running example: Generates H program "out.h".
python torch2h.py examples/mnist/main.py
- Make realistic weights by learning network. Current weights are just randomly generated floats.
- Add more benchmarks and functions accordingly.
This program has built on PyTea.