(x, global_sharding)

(*trees, bdims=None)

(self)

(mesh, partition_rules)

(stream)

(self, carry, policy, length, training, single=False)

(self, batch_size)

(last, term, rew, val, boot, disc, lam)

(self, event)

(config, index, **overrides)

(x)

(self)

(carry, stream)

(x, ts=None, inverse=False, maxlen=4096)

(self, action)

(self, carry, inp, reset)

(ax, xticks=4, yticks=4, grid=(1, 1), logx=False, darker=False)

(x)

(self, carry, data)

(self, key, uprob)

(self, name)

(self, action)

(self, data, is_online, is_first)

(self, seqs, start, stop)

(self, uprobs, bfactor=16)

(self)

(self)

(self, text)

(self, token)

(self)

(self, actions)

(self)

(self, name)

(self, logdir, train_ratio, eval_envs)

(self, logdir)

(self, index)

(self, index)

( self, lr: float = 4e-5, agc: float = 0.3, eps: float = 1e-20, beta1: float = 0

(self, name)

(self, value, mask)

(self, obs)

(self, time_step)

(self)

(self)

(self)

(self, image)

(self, tree, rng, bfactor=16)

(self, mode)

(self, chunkid, index, values)

(self, shape, dtype)

(port, replicas)

(self, policy, step, episode)

(self, counts, depth)

(self)

(self, flat)

(self, grads, finite)

(self, length, reward, info)

(df, args)

(self, x)

(params, compile=True)

(name, df)

(df, args)

(shape)

(x, prob, training)

(*args, inner=fn)

(data)

( act, rew, con, policy, value, slowvalue, retnorm, valnorm, advnorm, update, contdisc=Tru

(self, batch_size)

(self, batch_size)

(self, batch_size)

(fig, names=None, reverse=False, adjust=False, **kwargs)

(args)

(sharding)

(self, event)

(args)

(argv=None)

(config, folder, mode='train')

(filename)

(df, stats, args)

(amount, cols=4, size=(3, 3), **kwargs)

(self)

(grouping)

(df)

(self, stepids, priorities)

(self, stepids, priorities)

(self, event)

( last, term, rew, boot, value, slowvalue, valnorm, update=True, slowreg=1.0, slowtar=True

(self, carry, data)

(params, partition_rules, mesh)

(self, batch, mode='train')

()

(replica, clockid, skip)

(self, entries, carry, nlast)

(self)

(self, clients=32, batch=16, workers=4)

(self)

(self)

(self, clients=32, batch=16, workers=4)

(self)

(x, global_sharding)