Developing code for optimal exploration in language models - View it on GitHub