Using float32 instead of float64 in netket

[wlwuphy]

I'm trying to use float32 to reduce memory usage. What do I need to do to change the default dtype to float32? I try "jax.config.update('jax_enable_x64', False)" in the beginning of my code but it seems not working.

[PhilipVinc]

NetKet sets jax.config.update('jax_enable_x64', True) when you import it.
setting NETKET_ENABLE_X64=0 before importing netket should disable double precision in jax
or setting jax.config.update('jax_enable_x64', False) AFTER importing netket should address it

[PhilipVinc]

Though do notice that I don't think you can get away with single precision easily. There are considerable issues in using single precision and I strongly do not recommend it.

[wlwuphy]

Thank you! You are right, it doesnt work at all.