Run:AI Python library
Public functional modules for Keras and TF
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 models elastic |
Gradient Accumulation | ga |
Gradient accumulation for Keras optimizers |
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 usepip3
for Python3)
Running The Tests
All tests (unit tests) can be run using the following command:
python -m unittest discover -s tests -v