Feature/laser cooler

[pkruyt]

Description

Closes # .

Checklist

Mandatory:

• [ x] I have added tests to cover my changes
• [ x] All the tests are passing, including my new ones
• I described my changes in this PR description

Optional:

• [x ] The code I wrote follows good style practices (see PEP 8 and PEP 20).
• I have updated the docs in relation to my changes, if applicable
• I have tested also GPU contexts