Skip to main content

Large-Scale Machine and Deep Learning in PyTorch.

Project description

PyBlaze

PyPi

PyBlaze is a high-level library for large-scale machine learning in PyTorch. It is engineered to cut obsolete boilerplate code while preserving the flexibility of PyTorch to create just about any deep learning model.

Features

Generally, PyBlaze provides an object-oriented approach to extend PyTorch's API. The core design objective is to provide an API both as simple and as extensible as possible. PyBlaze's features include the following:

  • Training and prediction loops with minimal code required and callback support.
  • Out-of-the-box multi-GPU support where not a single additional line of code is required.
  • Intuitive multiprocessing by providing easy for-loop vectorization.
  • Modules and functions missing in PyTorch.

Currently, PyBlaze only provides means for running training/inference on a single machine. In case this is insufficient, you might be better off using PyTorch's distributed package directly.

It must be emphasized that PyBlaze is not meant to be a wrapper for PyTorch as Keras is for TensorFlow - it only provides extensions.

Installation

PyBlaze is available on PyPi and can simply be installed as follows:

pip install pyblaze

Quickstart

An introduction to PyBlaze is given as a tutorial training an image classifier. It can be found in the documentation's guide section.

License

PyBlaze is licensed under the MIT License.

Project details


Download files

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

Source Distribution

pyblaze-1.0.0.tar.gz (32.6 kB view hashes)

Uploaded Source

Built Distribution

pyblaze-1.0.0-py3-none-any.whl (46.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page