Maratyszcza/NNPACK

Integration with deep learning frameworks

Open

#1 opened on Mar 24, 2016

View on GitHub
 (21 comments) (0 reactions) (0 assignees)C (321 forks)github user discovery
help wanted

Repository metrics

Stars
 (1,707 stars)
PR merge metrics
 (No merged PRs in 30d)

Description

NNPACK needs to be integrated into deep learning frameworks to deliver performance benefits to end-users. If you would like to work on such integration, please comment here.

@ajtulloch contributed basic integration of NNPACK's convolutional layers into Caffe, which is now available at Maratyszcza/caffe-nnpack. This project is not complete: it only exposes NNPACK's training-optimized forward propagation in convolution layers to Caffe, but it is a good starting point.

Contributor guide

Research direction
Familiarize yourself with the existing integration in caffe nnpack (https://github.com/Maratyszcza/caffe nnpack). Identify which deep learning frameworks (e.g., TensorFlow, PyTorch) would benefit from NNPACK integration. Start by understanding the NNPACK API and how it accelerates convolutions. Then look into the build systems of the target framework to add NNPACK as a dependency. Finally, implement the integration and test performance improvements.
Tech stack
c
Domain
machine learningai
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
4
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
Over 1 week
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Mostly clear
Prerequisites
C programmingDeep learning frameworksGit
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
20