Developer material

TensorFlow

Type: Framework

Google framework with optimized 32-bit and 8-bit NEON routines for Arm CPUs.
Includes an OpenCL backend supporting Arm Mali GPU via SYCL.

TensorFlow Arm setup guide

PyTorch

Type: Framework

Facebook framework with optimized routines for Arm CPUs with Neon.

MxNet

Type: Framework

Apache framework with optimized routines for Arm CPUs.
Open AI version optimized with the Arm Compute Library for CPUs and GPUs.

MxNet GitHub

Microsoft Cognitive Toolkit

Type: Framework

Microsoft framework with optimized routines for Arm CPUs with Neon.

Caffe

Type: Framework

Berkley Vision and Learning Centre framework with several options for running on Arm.
CaffeOnACL is a version optimized with the Arm Compute Library for CPUs and GPUs.

CaffeOnACL GitHub

Keras

Type: API

A high-level neural networks API, written in Python and capable of running on top of several framework backends including TensorFlow and Microsoft Cognitive Toolkit (CNTK).

PlaidML

Type: Keras backend

Vertex.ai OpenCL framework compatible with Keras API with support for Arm Mali GPU.

Keil MDK

Arm Keil MDK is the most comprehensive software development solution for Arm-based microcontrollers and includes all components that you need to create, build, and debug embedded applications.

Arm NN

Arm NN is the Arm inference engine. Arm NN is designed to optimally run networks that are trained on popular frameworks, like TensorFlow and Caffe, on Arm IP.

Arm Compute Library

Arm Compute Library optimizes low-level functions for computer vision and machine learning. Arm Compute Library focuses on Convolutional Neural Networks for 32-bit float and 8-bit integers across an array of Arm CPUs and GPUs.

CMSIS-NN

CMSIS-NN is the Arm library of efficient neural network kernels for Arm Cortex-M CPUs.

DSP extensions for Cortex-M

Cortex-M processors with DSP provide a high level of signal processing and integer performance, while maintaining the energy-efficiency and ease-of-use hallmarks of the Cortex-M family.

DSP extensions for Cortex-R

This instruction set for Cortex-R processors includes enhanced DSP instructions that improve execution performance for arithmetic operations.