Filippo Vicentini

Wrong parameter names when nesting Modules within flax transformations

3

Hi, I have a complex case where I nest different submodules inside each other, which results in what I think is a wrong parameter name. MWE: ```python import jax import...

Binary app error: Could not find a version that satisfies the requirement

1

Hello, I tried to setup a project using hatch/pyapp, but after building the project with pyapp (which completes successfully) I keep getting the following error. ```python ERROR: Could not find...

Set replicate-sharding for parameters of MCState

1

Also sets the replicated sharding on the nkjax.PRNGKey, as this is what the contract says for it. Part of the reason for this is because recently we realised that when...

Allow custom constraints on SpinOrbitalFermions

1

This PR makes use of the interface defined in #1805 to implement the constraint of the population on subsections of SpinOrbitalFermions, removing the internal Fock Hilbert space. While this change...

When running under MPI we serialise only the sampler_state on the first rank, and ignore the one on the other ranks. This is wrong. When you load it, you get...

I don't want to make the switch automatic, but giving a more informative message to the user whether we detect it enabled or not might be useful for debugging configurations...