xpag.samplers.jax_sampler.JaxSampler#

class JaxSampler(*, seed=None)#

Bases: Sampler

Methods

`init`
`sample`	Return a batch of transitions

sample(buffer_state, batch_size)#

Return a batch of transitions