Shengdu Chai issues

Results 3 issues of


                                            Shengdu Chai

DeepSolid code optimization failure for ground state energy calculation of 222 bcc Hydrogen

**Description:** I am experiencing issues with the DeepSolid code while calculating the ground state energy of 222 supercell, bcc Hydrogen. Despite trying multiple hyperparameters, such as learning rate, MCMC steps,...

Unexpected change in parameter registration from 'Orphan' to 'Generic' causes training failure in kfac_jax 0.0.7

When upgrading from `kfac_jax==0.0.6` (with `jax==0.4.35`) to `kfac_jax==0.0.7` (with `jax==0.4.36`), some parameters that were previously registered as **`Orphan`** are now being registered as **`Generic`**. This change leads to a training...

TypeError: unhashable type: 'Literal'

### Issue Report: **TypeError with Constant Multiplication in network When Using kfac_jax** #### **Summary** When using `kfac_jax` with a network, introducing a scaling factor (e.g., `geo_scale`) for lattice parameters in...