Handle the new JAX bind-params API#42

gcarleo wants to merge 1 commit intomicrosoft:mainmicrosoft/folx:mainfrom

netket:codex/fix-get-bind-params-upstreamnetket/folx:codex/fix-get-bind-params-upstreamCopy head branch name to clipboard