Rubiks Cube Gym

Rubiks Cube Gym

After my internship ended prematurely due to the COVID-19 pandemic, I decided to use my newfound free time to learn new things and hone my skills. On the programming side of things, this has been reinforcement learning. I have been fascinated by reinforcement learning and OpenAI since I first read about it. OpenAI has led to some significant breakthroughs in recent years. From Gym, which standardised environments and can be used with all major RL libraries to the OpenAI Five, a team of five individual bots which beat the reigning champions OG at DotA2 to GPT-2 which completely revolutionised the NLP field.

I am also a big fan of the Rubik’s Cube and various twisty puzzles. Combining these two, I have created an OpenAI Gym environment for twisty puzzles. Currently, it supports the 2x2 Rubik’s Cube with more puzzles such as the Pyraminx and Skewb to come in the near future. I had initially intended to create one for the 3x3 Rubik’s Cube which everyone knows and loves, but the number of possible states is far too large, 43 quintillion or 4.3 x 10^19, making it impractical to use.

The 2x2 Rubiks Cube is available in three different environments, each for different reward methods. These include layer by layer, a beginners method and the Ortega method, which is commonly used by speedcubers.

This project has been a fantastic learning experience, and I plan to use it to do some Reinforcement Learning using it in the near future(stay tuned for that).

The code is available on my Github along with some more technical information in the README file.

I would love to see how anyone ends up using it. As always, I am open to suggestions and feedback.

Photo Credits: Gerwin Sturm

Kavish Hukmani

Kavish Hukmani

TODO Add a witty bio

rss facebook twitter github gitlab youtube mail spotify lastfm instagram linkedin google google-plus pinterest medium vimeo stackoverflow reddit quora quora