/sparse-reward-buffers

batched sparse reward buffers in JAX

Primary LanguagePythonApache License 2.0Apache-2.0

sparse-reward-buffers

Implementations of batched sparse reward buffers in JAX.

Sparse reward buffers are useful when a reward is determined only at the end of an episode but needs to be associated with each sample in the episode trajectory.