simple history encoder for brax#651

samindaa wants to merge 2 commits intogoogle:maingoogle/brax:mainfrom

samindaa:history-encodersamindaa/brax:history-encoderCopy head branch name to clipboard