MLGO is tensor library for machine learning in pure Golang that can run on MIPS.
The machine learning part of this project refers to the legendary ggml.cpp framework.
pip install -r requirements.txt
- Train the AI model. See
examples/mnist/trainning/mnist.ipynb
- Convert the AI model into GGML using
examples/mnist/convert-h5-to-ggml.py
- Build the AI inference engine for MIPS
cd examples/mnist_mips && ./build
``
Support LLaMA 1 and LLaMA 2 (7B and 13B).
Please refer to examples/llama/README.md
and download the model first.