Fixed convnext tests and nnx warnings#135

jenriver merged 2 commits intojax-ml:mainjax-ml/bonsai:mainfrom

vfdev-5:fix-deprwarnings-nnx-state-to-pure-statevfdev-5/bonsai:fix-deprwarnings-nnx-state-to-pure-stateCopy head branch name to clipboard