What's included

WML Accelerator includes these components and frameworks.

Components

The following components are included with WML Accelerator:

Deep learning frameworks

The following frameworks are included with WML Accelerator for use with IBM Spectrum Conductor Deep Learning Impact. Some frameworks are available via the user interface. Others can only be used through the command line interface. For details, see Using frameworks via command line interface

Table 1. Included frameworks
Framework Version Python 2.7 support Python3.6 support Supported via user interface Supported via command line Training options
TensorFlow 1.14 Yes Yes X X
  • Single node training
  • Distributed training
Keras 2.2.4 Yes1 Yes1   X
  • Single node training
  • Distributed training
  • Elastic distributed training
IBM Caffe 1.0 Yes No X X
  • Single node training
PyTorch 1.1.0 Yes Yes X X
  • Single node training
  • Elastic distributed training
snap.ml 1.3.0 Yes1 Yes   X  
1: Support depends on what the user sets up. See Using frameworks via command line interface for details.      

Technology preview packages

The Program includes the following technology preview packages:
  • TFLMSv1
  • SnapML: snap-ml-spark library
  • pai4sk: Decision Tree and Random Forest algorithms from SnapML
  • RAPIDS.AI: cuML and cuDF
Before using technology preview packages, read the information in Technology Preview Code.