carletonai/CuMind

A JAX-based CuMind is a JAX-based RL framework inspired by Google DeepMind. It combines Monte Carlo Tree Search (MCTS) with a learned model to achieve superhuman performance in complex domains without prior knowledge of their rules.