/recipes-ml

Yocto recipes for machine learning libraries (mainly for Zynq-7000 & Zynq MPSoC devices)

Primary LanguageBitBake

recipes-ml

  • Yocto recipes for machine learning libraries

arm-computelibrary: ARM ComputeLibrary (v24.06 & v23.08)

armnn: Arm NN ML Software (v24.05)

caffe: Caffe v1.0

oneDNN: oneAPI Deep Neural Network Library (v3.4)

  • Note: MPSoC (64-bit processor) only

PyTorch: PyTorch (v2.3.1 & v1.13.1)

PyXIR: Xilinx PyXIR (v0.3.5)

tflite: TensorFlow Lite (TF v1.15.5)

tflite2: TensorFlow2 Lite (TF v2.14.0)

TorchVision: TorchVision (v0.18.1)

TVM: Apache TVM (v0.13.0)


Some notes on compiling PyTorch

  • Ample amount of RAM will be necessary (preferably >= 32GB)