TensorFlow and Keras

The ray.train.tensorflow module runs distributed TensorFlow training. The Keras callback lives in ray.train.tensorflow.keras.

TensorflowTrainer	A Trainer for data parallel Tensorflow training.
TensorflowConfig	TensorflowConfig()
prepare_dataset_shard	A utility function that overrides default config for Tensorflow Dataset.
ReportCheckpointCallback	Keras callback for Ray Train reporting and checkpointing.