desy-ml / cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
https://cheetah-accelerator.readthedocs.io
GNU General Public License v3.0
33 stars 13 forks source link

Fix issues when running on a machine with CUDA GPUs #97

Closed jank324 closed 10 months ago

jank324 commented 11 months ago

Bug fix for #87

Most important TODOs before merging:

Additional TODOs: