A curated collection of tools, frameworks, and libraries for Edge AI development.
Frameworks & Runtimes
TensorFlow Lite
Google’s lightweight solution for mobile and embedded ML deployment.
- Use case: Mobile apps, microcontrollers, IoT devices
- Platforms: Android, iOS, Linux, microcontrollers
- Website: tensorflow.org/lite
ONNX Runtime
Microsoft’s cross-platform inference engine with broad hardware support.
- Use case: Cross-platform deployment, hardware acceleration
- Platforms: Windows, Linux, macOS, Android, iOS
- Website: onnxruntime.ai
PyTorch Mobile
Meta’s solution for deploying PyTorch models on mobile devices.
- Use case: Mobile apps with PyTorch workflows
- Platforms: Android, iOS
- Website: pytorch.org/mobile
Development Platforms
Edge Impulse
End-to-end platform for building, deploying, and managing edge ML.
- Use case: Rapid prototyping, production deployment
- Features: Data collection, model training, optimization, deployment
- Website: edgeimpulse.com
STM32Cube.AI
STMicroelectronics’ tool for deploying neural networks on STM32 MCUs.
- Use case: Industrial IoT, consumer electronics
- Features: Model analysis, code generation, benchmarking
- Website: st.com/stm32cubeai
Model Optimization
TensorFlow Model Optimization Toolkit
Comprehensive toolkit for quantization, pruning, and clustering.
- Techniques: Post-training quantization, QAT, pruning, weight clustering
- Website: tensorflow.org/model_optimization
Benchmarking
MLPerf Tiny
Industry-standard benchmark for measuring ML performance on embedded devices.
- Use case: Hardware comparison, competitive benchmarking
- Workloads: Keyword spotting, visual wake words, image classification, anomaly detection
- Website: mlcommons.org/en/inference-tiny
Visualization & Debugging
Netron
Visualizer for neural network models supporting all major formats.
- Formats: ONNX, TensorFlow, PyTorch, TFLite, and more
- Website: netron.app
Hardware-Specific SDKs
NVIDIA TensorRT
High-performance inference optimizer for NVIDIA GPUs.
- Platforms: Jetson, data center GPUs
- Website: developer.nvidia.com/tensorrt
Google Coral
Tools for Google’s Edge TPU accelerator.
- Platforms: Coral Dev Board, USB Accelerator
- Website: coral.ai