connorwatts / rubix Goto Github PK
View Code? Open in Web Editor NEWDeep Reinforcement Learning based Rubik’s Cube solver written in JAX, Haiku and RLax. Developed a custom environment supporting cubes of variable size. Currently developing support for DQN, QR-DQN and PPO.
License: MIT License