HF RL Explorer

I want MolecularHamiltonian.local energy(ansatz) to be available on an already constructed…

I want MolecularHamiltonian.local energy(ansatz) to be available on an already constructed…: a task in MiMo-V2.6-RL-oss: Agentic RL Environments (MiMo RL release). I want MolecularHamiltonian.local energy(ansatz) to be available on an already constructed MolecularHamiltonian. It should return a…

The task

I want MolecularHamiltonian.local_energy(ansatz) to be available on an already constructed MolecularHamiltonian. It should return a callable with signature (rng: Optional[KeyArray], params: Params, phys_conf: PhysicalConfiguration) -> tuple[Energy, Stats]. In a session like…

Part of XiaomiMiMo/MiMo-V2.6-RL-oss.