Implementation of Algorithm Distillation
Implement Algorithm Distillation from the paper In Context Reinforcement Learning with Algorithm Distillation, where a single transformer model is trained to in-context learn the source RL algorithm to solve various RL tasks. The project is implemented on the Darkroom environment.