Skip to content

TinyLlama on Cortex-M55 using CMSIS-DSP and Helium vector instructions

License

Notifications You must be signed in to change notification settings

christophe0606/MLHelium

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

TinyLlama on Cortex-M55 using CMSIS-DSP and Helium vector instructions

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published