We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b33f130 commit 2e4046cCopy full SHA for 2e4046c
tensorflow_probability/python/experimental/distribute/sharded.py
@@ -46,7 +46,7 @@ def lp_fn(self, x):
46
class Sharded(distribution_lib.Distribution):
47
"""A meta-distribution meant for use in an SPMD distributed context.
48
49
- `Sharded` is a meta-distribution enables distributions to be used in SPMD
+ `Sharded` is a meta-distribution that enables distributions to be used in SPMD
50
programs. A `Sharded` distribution represents a random variable that has
51
been split across a set of devices. The number of shards is the number of
52
devices in the current TensorFlow DistributionStrategy or the provided JAX
0 commit comments