extensions in rllib