desy-ml/cheetah

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.