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
32 stars 13 forks source link

Bmad import does not pass `device` #195

Closed jank324 closed 3 months ago

jank324 commented 3 months ago

Right now it's not possible to set the device when importing a lattice from Bmad.