code: def set_velocities(self, velocities):
          
          assert len(velocities) == len(self.mol)
          self.params["velocity"] = velocities
docs: :param velocities (au): list of list of atom velocities 
          :return: