fix(tests): update tolerance levels and PRNGKey
usage for improved test stability
#3751
Loading
PRNGKey
usage for improved test stability
#3751