Skip to main content

Run:AI Python library

Project description

Run:AI Python library

Public functional modules for Keras, TF and PyTorch



CircleCI is used for CI system: CircleCI


This library consists of a few pretty much independent submodules:

Module Name Info
Elastic elastic Make Keras and PyTorch models elastic
Gradient Accumulation ga Gradient accumulation for Keras and PyTorch optimizers
HPO hpo Hyperparameter optimization assistance
Model Parallelism mp Model-parallelism support for Keras builtin layers
Auto Profiler profiler Export timeline of TF/Keras models easily
Reporter reporter An interface to send metrics and parameters to Promethues Push Gateway

Getting Started


Install the runai Python library using pip using the following command:

pip install runai

Make sure to use the correct pip installer (you might need to use pip3 for Python3)

Running The Tests

All tests (unit tests) can be run using the following command:

python -m unittest discover -s tests -v

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

runai-0.4.1.tar.gz (18.8 kB view hashes)

Uploaded source

Built Distribution

runai-0.4.1-py2-none-any.whl (30.2 kB view hashes)

Uploaded py2

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