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