Tags: poets-ai/elegy
Tags
Distributed API (#194) * prepare core model * basic distributed support * basic distributed * DistributedStrategy class * use current strategy when calling to_local * fix low-level-api example * change _losses_and_metrics from lambda to Hashable * add distributed flag to elegy mnist_conv example * add distributed flag to flax and haiku mnist_conv examples * update deps
PreviousNext