fix(tests): update tolerance levels and PRNGKey
usage for improved test stability
#3764
This job was skipped
Loading
PRNGKey
usage for improved test stability
#3764