GitLab

* Make training benchmark print a warning if not synchronizing, and add a sync before function also

* Take triu_index out since it does not run

* slightly increase default precision

* flake8
Co-authored-by: Ignacio Pickering <ign.pickering@gmail.com>