Update README.md

README.md

@@ -88,6 +88,15 @@ wf = FlaxAutoModel.from_pretrained("nqs-models/j1j2_square_10x10", trust_remote_
 
 Use `revision="symm_trxy_ising"` for a wavefunction including also the point group and the sz inversion symmetries.
 
+### Extract hidden representation
+
+The hidden representation associated to the input batch of configurations can be extracted as:
+
+```python
+wf = FlaxAutoModel.from_pretrained("nqs-models/j1j2_square_10x10", trust_remote_code=True, return_z=True)
+
+z = wf.__call__(wf.params, samples)
+```
 
 #### Training Hyperparameters