Implementation of a transformer for reinforcement learning using `x-transformers` - View it on GitHub