Handle the new JAX bind-params API#41

gcarleo wants to merge 3 commits intomicrosoft:mainmicrosoft/folx:mainfrom

netket:codex/fix-get-bind-paramsnetket/folx:codex/fix-get-bind-paramsCopy head branch name to clipboard