Fixes compatibility issues with jax v0.2
Pre-release
Pre-release
Merge pull request #63 from aboucaud/remove-duplicate-sqrtk Remove duplicate `sqrtk` accessor
Merge pull request #63 from aboucaud/remove-duplicate-sqrtk Remove duplicate `sqrtk` accessor