1–6 Jun 2025
Taipei International Convention Center (TICC)
Asia/Taipei timezone

Updates to the differentiable accelerator simulation code Cheetah

WEPS034
4 Jun 2025, 16:00
2h
Exhibiton Hall A _Salmon (TWTC)

Exhibiton Hall A _Salmon

TWTC

Poster Presentation MC5.D11 Code Developments and Simulation Techniques Wednesday Poster Session

Speaker

Andrea Santamaria Garcia (Karlsruhe Institute of Technology)

Description

The design and operation of modern accelerators demand advanced simulation tools capable of addressing complex challenges. Differentiable simulations are particularly valuable, as they enable gradient-based optimization techniques that significantly reduce computational costs and efficiently tackle high-dimensional problems. The PyTorch-based simulation code Cheetah was developed to combine high-speed, differentiable simulations with seamless integration into machine learning workflows. In this work, we present recent updates to Cheetah, developed collaboratively by DESY, KIT, SLAC, and LBNL, which extend its capabilities and enhance its performance. Key advancements include support for vectorized execution, enabling simultaneous simulations across large parameter spaces; the addition of space charge modeling and higher-order transfer maps for more accurate beam dynamics; and expanded support for multiple particle species and additional accelerator components, broadening its applicability to other systems. By enabling faster, more precise, and scalable simulations, Cheetah is poised to become a valuable tool for meeting the growing demands of the accelerator physics community.

Region represented Europe
Paper preparation format LaTeX

Author

Chenran Xu (Karlsruhe Institute of Technology)

Co-authors

Jan Kaiser (Deutsches Elektronen-Synchrotron DESY) Christian Hespe (Deutsches Elektronen-Synchrotron DESY) Annika Eichler (Deutsches Elektronen-Synchrotron DESY) Andrea Santamaria Garcia (Karlsruhe Institute of Technology) Ryan Roussel (SLAC National Accelerator Laboratory) Juan Pablo Gonzalez-Aguilera (University of Chicago) Remi Lehe (Lawrence Berkeley National Laboratory) Grégoire Charleux (Lawrence Berkeley National Laboratory)

Presentation materials

There are no materials yet.