Entity Gym compatible ragged batch transformer implementation. - View it on GitHub