VAE tag Agents

For a Research Project I investigated and implemented a Variational Auto-Encoder (VAE),a deep learning architecture, in a Reinforcement learning Environment. The Green dot is trained to stay within the box and the two red dots are trained to push the green dot out of the box. Technologies used Pytorch, google Colab Pro, AI gym.