Run:AI Python library

Public functional modules for Keras, TF and PyTorch

Info

Status

CircleCI is used for CI system:

Modules

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

Installing

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