[D] Run Pytorch model inference on Microcontroller

This page summarizes the projects mentioned and recommended in the original post on /r/MachineLearning

Our great sponsors
  • WorkOS - The modern identity platform for B2B SaaS
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • SaaSHub - Software Alternatives and Reviews
  • tinyengine

    [NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning; [NeurIPS 2022] MCUNetV3: On-Device Training Under 256KB Memory (by mit-han-lab)

  • TinyEngine from MCUNet. Looks great, targeting ARM CM4.

  • CMSIS-NN

    CMSIS-NN Library

  • CMSIS-NN. ARM centric. Examples. They also have an example for a pytorch to tflite converter via onnx

  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

    WorkOS logo
  • ML-examples

    Arm Machine Learning tutorials and examples

  • CMSIS-NN. ARM centric. Examples. They also have an example for a pytorch to tflite converter via onnx

  • TinyMaix

    TinyMaix is a tiny inference library for microcontrollers (TinyML).

  • TinyMaix. Very minimalistic, can also be used on RISC-V

  • nnom

    A higher-level Neural Network library for microcontrollers.

  • nnom

  • ai8x-synthesis

    Quantization and Synthesis (Device Specific Code Generation) for ADI's MAX78000 and MAX78002 Edge AI Devices

  • MAX7800X Toolchain and Documentation (proprietary) This is a proprieteray toolchain to deploy models to the MAX78000 edge NN devices.

  • MaximAI_Documentation

    START HERE: Documentation for ADI's MAX78000 and MAX78002 Edge AI devices

  • MAX7800X Toolchain and Documentation (proprietary) This is a proprieteray toolchain to deploy models to the MAX78000 edge NN devices.

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo
  • deepC

    vendor independent TinyML deep learning library, compiler and inference framework microcomputers and micro-controllers

  • DeepC. Open source version of DeepSea. Very little activity, looks abandoned

  • onnx2c

    Open Neural Network Exchange to C compiler.

  • onnx2c - onnx to c sourcecode converter. Looks interesting, but also not very active.

  • cONNXr

    Pure C ONNX runtime with zero dependancies for embedded devices

  • cONNXr - framework with C99 inference engine. Also interesting and not very active.

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts