Merge branch 'trunk' into bengioe-qol-and-docs

src/gflownet/envs/graph_building_env.py

@@ -626,6 +626,7 @@ def detach(self):
 
     def to(self, device):
         self.dev = device
+        self.raw_logits = [i.to(device) for i in self.raw_logits]
         self._masked_logits = [i.to(device) for i in self._masked_logits]
         self.batch = [i.to(device) for i in self.batch]
         self.slice = [i.to(device) for i in self.slice]