cloudpickle
psutil
pyyaml
six
cffi>=1.4.0

[all-frameworks]
tensorflow
tensorflow-gpu
keras!=2.0.9,!=2.1.0,!=2.1.1,>=2.0.8
torch
mxnet>=1.4.1
h5py>=2.9
numpy
petastorm>=0.9.0
pyarrow>=0.15.0
pyspark>=2.3.2

[keras]
keras!=2.0.9,!=2.1.0,!=2.1.1,>=2.0.8

[mxnet]
mxnet>=1.4.1

[pytorch]
torch

[spark]
h5py>=2.9
numpy
petastorm>=0.9.0
pyarrow>=0.15.0
pyspark>=2.3.2

[tensorflow]
tensorflow

[tensorflow-cpu]
tensorflow-cpu

[tensorflow-gpu]
tensorflow-gpu
