Replies: 1 comment 1 reply
-
This comment has been outdated since a few years. We upstreamed all of our changes to flax, so you can safely use flax itself. |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I have seen conflicting advice on whether to use
import flax.linen as nn
orimport netket.nn as nn
if my network involves complex numbers.On the documentation page for Variational State, there is this warning which suggests using the repackaged netket.nn if my model involves complex parameters:
However, when I try to replace the import of nn on my local installation (version 3.13.0), I get this warning message when I initialize some codes. It seems in the source code, the nk.nn.Module and nk.nn.compact is deprecated:
What is the current best practice for complex models? Should I use flax.linen or netket.nn?
Beta Was this translation helpful? Give feedback.
All reactions