A Neural Network Toolkit.
Clone or download
odashi Merge pull request #202 from primitiv/feature/hierarchy
Move primitiv::Device::DeviceType to primitiv::DeviceType.
Latest commit 6748b5b Aug 12, 2018

README.md

c++ backend os license

  • Branch master: Build status (master) Documentation Status (master)
  • Branch develop: Build status (develop) Documentation Status (develop)

primitiv

A Neural Network Toolkit.

Features

  • Dynamic and incremental graph construction (a.k.a. "define-by-run" style)
  • On-demand memory allocation
  • Automatic minibatch broadcasting
  • Mostly device-independent
  • Simple usage

Languages

This repository contains only the core C++ library and its C APIs. Some bindings for other programming languages (e.g., Python) can be found in the official repository.

Prerequisites

  • CMake 3.1.0 or later
  • C++11 compiler (GCC, Clang)
  • (optional) Google Test
    • Required only when -DPRIMITIV_BUILD_TESTS=ON.
  • (optional) CUDA 8.0 or later/cuDNN 7.0 or later
    • Required only when -DPRIMITIV_USE_CUDA=ON
  • (optional) OpenCL 1.2/OpenCL C++ binding v2
    • Required only when -DPRIMITIV_USE_OPENCL=ON

Documentation

Contact

This project is supported by ASTREC in NICT.