fix: put tensors on device during creation

[ImahnShekhzadeh]

Fixes #102

Comment: In the NB model-comparison-plotting.ipynb, I left things as they are, since the device is explicitly chosen to be the CPU, i.e. no CPU-GPU overhead exists.

PS: @kilianFatras The tensors in the tutorials are also created on the corresponding device now, since I fixed that before you posted your comment :slightly_smiling_face:

Before submitting

• [x] Did you make sure title is self-explanatory and the description concisely explains the PR?
• [x] Did you make sure your PR does only one thing, instead of bundling different changes together?
• [x] Did you list all the breaking changes introduced by this pull request?
• [x] Did you test your PR locally with pytest command?
• [x] Did you run pre-commit hooks with pre-commit run -a command?

[review-notebook-app[bot]]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[codecov[bot]]

Codecov Report

Attention: 6 lines in your changes are missing coverage. Please review.

Comparison is base (a8f614b) 35.92% compared to head (f8c6d1f) 35.92%.

Files	Patch %	Lines
runner/src/models/cfm_module.py	20.00%	4 Missing :warning:
runner/src/models/components/plotting.py	0.00%	2 Missing :warning:

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #103 +/- ## ======================================= Coverage 35.92% 35.92% ======================================= Files 67 67 Lines 7399 7399 ======================================= Hits 2658 2658 Misses 4741 4741 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.