(self, carry, data)

(condition, xs, ys)

(env, config)

(self)

(self)

(self)

(self)

(self)

(self)

(self)

(self, policy, steps=0, episodes=0)

(self, step=None, skip=None)

(self, step=None, skip=None)

(self, obs, inventory)

(self, obs, inventory=None)

(self, x, c, k)

(self, x)

(self, lossfn, *args, has_aux=False, **kwargs)

(self, x, update)

(self, *args, **kwargs)

(self, *args, **kwargs)

(self, shape, dtype=jnp.float32, fshape=None)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x, mask=None, ts=None, training=True)

(self, xs)

(self, xs, bshape)

(self, x)

(self, x, mask=None, ts=None, training=True)

(self, carry, inputs, resets, single=False)

(self, x, bdims)

(self, x)

(self, x)

(self, carry, obs, reset, training, single=False)

(self, carry, feat, reset, training, single=False)

(self, key)

(self, key)

(self, key)

(self, key)

(self, key)

(self, other)

(self, name)

(self, name)

(self, name)

(self, name)

(self)

(self, seed=0)

(self, uprobs, seed=0)

( self, exponent=1.0, initial=1.0, zero_on_sample=False, maxfrac=0.0, branching=16, seed=0)

(self, selectors, fractions, seed=0)

(self, branching=16, seed=0)

(self, parent=None)

(self, key=None, uprob=None)

(self, make_env_fns, parallel=True, **kwargs)

(self, obs_space, act_space, config)

(self, obs_space, act_space)

( self, length, capacity=None, directory=None, chunksize=1024, online=False, selector=None, save_w

(self, nextfn, *args, **kwargs)

(self, source, transform=None, amount=1)

( self, source, length, consec, prefix=0, strict=True, contiguous=False)

(self, sources)

(self, source, fn, *args, **kwargs)

(self, sources, weights, seed=0)

(self, samples_per_insert, tolerance, minsize)

(self, size=1024)

(self, every, first=False)

(self, every, first=False)

(self, env)

(self, env, duration, reset=True)

(self, env, repeat)

(self, env, key='action', low=-1, high=1)

(self, env, key='action')

(self, env)

(self, env)

(self, env, key='action', bins=5)

(self, env, size=(64, 64))

(self, env, horizon)

( self, ctor, exceptions=(Exception,), window=300, maxfails=2, wait=20)

( self, name, repeat=4, size=(84, 84), gray=True, noops=0, lives='unused', sticky=True, actions='a

( self, name, repeat=1, size=(64, 64), camera=-1, again=False, termination=False, weaker=1.0)

(self, task, size=(64, 64), logs=False, logdir=None, seed=None)

(self, env, obs_key='image', act_key='action', **kwargs)

( self, env, repeat=1, size=(64, 64), proprio=True, image=True, camera=-1)

(self, task, size=(64, 64), length=100)

(self, task, length=10000)

(self, task)

(self, task, size=(64, 64), resize='pillow', **kwargs)

(self, task, *args, **kwargs)

(self, *args, **kwargs)

(self, *args, **kwargs)

(self, item, once=0, repeated=0)

(self, scale=0.01)

( self, actions, repeat=1, size=(64, 64), break_speed=100.0, gamma=10.0, s

(self, resolution=(64, 64), break_speed=50)

(self, multiplier=1.0)

(self, env, obs_key='observation', act_key='action')