This repo has code to create a matmul model in torch then convert it to rknn model and run it on rockchip npu - View it on GitHub