Skip to content

Latest commit

 

History

History
22 lines (10 loc) · 1.07 KB

README.md

File metadata and controls

22 lines (10 loc) · 1.07 KB

MLHelium

Experiment with ML using Helium vector instructions. It relies on CMSIS-DSP but a few additional kernels may be provided by this library when not available in CMSIS-DSP.

It is NOT a new ML framework. It is just some examples to demonstrate that for simple networks, CMSIS-DSP may be enough (but requires more work).

This library is only focusing small networks with the goal of requiring the minimum of dependencies to run a model.

If you need fully quantized kernels and/or use of an Arm NPU for Cortex-M, please use TensorFlow Lite for Micro and CMSIS-NN.

The model must be written in C (by calling the required kernels). No automatic conversion from TensorFlow or PyTorch is provided.

A script is provided to make it easier to export the weights from the Python and use them from the C.

LICENSE is Apache except for some examples that have been adapted from other github repositories and may use other licences (like MIT).

Here is a video of the TinyLLama at 50 MHz on Cortex-M55:

TinyLLama_int8_f32_an552_final.mp4