#178 · Primary category: Deep Learning Frameworks
tensorpack
A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility
Project last updated:08/06/23
GitHub Stars
6.3K
Forks
1.8K
Contributors
58
License
Apache-2.0
Why we included this project
Tensorpack exists to make graph-mode TensorFlow train faster. It is a training interface rather than another layer of model wrappers, and it ships off-the-shelf multi-GPU and distributed data-parallel strategies. The dataflow system parallelizes pure-Python preprocessing so your data pipeline does not become the bottleneck, a trade-off the project defends explicitly against tf.data. The repo also includes faithful re-implementations of published papers instead of toy MNIST demos, which gives researchers a solid starting point for baselines. If you work in TensorFlow and need both speed and flexibility, this is a mature option.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.