Skip to main content
This is a pre-production deployment of Warehouse. Changes made here affect the production instance of PyPI (
Help us improve Python packaging - Donate today!

Neural Network Toolbox on TensorFlow

Project Description

A neural net training interface based on TensorFlow.

See some examples to learn about the framework. Everything runs on multiple GPUs, because why not?

Speech / NLP:

Examples are not only for demonstration of the framework – you can train them and reproduce the results in papers.


It’s Yet Another TF wrapper, but different in: 1. It’s not a model wrapper. + There are already too many symbolic function wrappers. Tensorpack includes only a few common models, but you can use any other model wrappers within tensorpack, such as sonnet/Keras/slim/tflearn/tensorlayer/….

  1. Focus on training speed.
    • Speed comes for free with tensorpack – it uses TensorFlow in the correct way. On various CNNs, it runs 1.5~1.7x faster than the equivalent Keras code.
    • Data-parallel multi-GPU/distributed training is off-the-shelf to use. It is as fast as Google’s official benchmark.
    • See tensorpack/benchmarks for some benchmark scripts.
  2. Focus on large datasets.
    • It’s painful to read/preprocess data through TF. tensorpack helps you load large datasets (e.g. ImageNet) in pure Python with autoparallelization. It also naturally works with TF Queues or
  3. Interface of extensible Callbacks. Write a callback to implement everything you want to do apart from the training iterations, and enable it with one line of code. Common examples include:
    • Change hyperparameters during training
    • Print some tensors of interest
    • Monitor GPU utilization
    • Send error rate to your phone

See tutorials to know more about these features.



  • Python 2 or 3

  • TensorFlow >= 1.0.0 (>=1.1.0 for Multi-GPU)

  • Python bindings for OpenCV (Optional, but required by a lot of features)

    pip install -U git+
    # or add `--user` to avoid system-wide installation.

    If you only want to use tensorpack.dataflow alone as a data processing library, TensorFlow is also optional.

Release History

Release History

This version
History Node


History Node


History Node


History Node


History Node


History Node


History Node


History Node


History Node


History Node


Download Files

Download Files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

File Name & Checksum SHA256 Checksum Help Version File Type Upload Date
tensorpack-0.6.0-py2.py3-none-any.whl (233.8 kB) Copy SHA256 Checksum SHA256 py2.py3 Wheel Oct 18, 2017
tensorpack-0.6.0.tar.gz (165.0 kB) Copy SHA256 Checksum SHA256 Source Oct 18, 2017

Supported By

WebFaction WebFaction Technical Writing Elastic Elastic Search Pingdom Pingdom Monitoring Dyn Dyn DNS Sentry Sentry Error Logging CloudAMQP CloudAMQP RabbitMQ Heroku Heroku PaaS Kabu Creative Kabu Creative UX & Design Fastly Fastly CDN DigiCert DigiCert EV Certificate Rackspace Rackspace Cloud Servers DreamHost DreamHost Log Hosting