Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

Module: tf.distribute.experimental

TensorFlow 2 version

Experimental Distribution Strategy library.


class CentralStorageStrategy: A one-machine strategy that puts all variables on a single device.

class CollectiveCommunication: Communication choices for CollectiveOps.

class MultiWorkerMirroredStrategy: A distribution strategy for synchronous training on multiple workers.

class ParameterServerStrategy: An asynchronous multi-worker parameter server tf.distribute strategy.

class TPUStrategy: TPU distribution strategy implementation.