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