We implemented a basic OpenAI Gym environment named the Rotatron that accepts a molecular graph (either at the atom or residue level) and an optional list of rotatable edges to optimize. If not provided, rotatable edges are inferred from the graph directly. Using the Rotatron as a parent class, we implemented three optimization environments that use different heuristics to evaluate a given conformation.