Zero

Zero is a general-purpose library for PyTorch users. Zero:

  • simplifies training loop, reproducibility, models evaluation and other typical Deep Learning tasks

  • provides a collection of “building blocks” and leaves code organization to you

  • can be used on its own or together with PyTorch frameworks such as Lightning, Ignite, Catalyst and others

NOTE: Zero is tested (and battle-tested in research projects), but the interface is not stable yet, so backward-incompatible changes in future releases are possible.

Overview

Installation

If you plan to use the GPU version of PyTorch, install it before installing Zero (otherwise, the CPU version will be installed together with Zero).

$ pip install libzero

Dependencies

  • Python >= 3.7

  • NumPy >= 1.17

  • PyTorch >= 1.6 (CPU or CUDA >= 10.1)

  • pynvml >= 8.0

  • tqdm >= 4.0

How to contribute

If your contribution includes pull requests, see CONTRIBUTING.md

Why “Zero”?

Zero aims to be zero-overhead in terms of mental overhead: solutions, provided by Zero, try to be as minimal, intuitive and easy to learn, as possible. Well, all these things can be pretty subjective, so don’t take it too seriously :wink:


Indices and Tables