JuTrack: A Julia package for auto-differentiable accelerator modeling and particle tracking

Journal Article · 06 January 2025 · Computer Physics Communications

DOI: https://doi.org/10.1016/j.cpc.2024.109497 · OSTI ID:2544245

Wan, Jinyu  ^[1]; Alamprese, Helena ^[1]; Ratcliff, Christian ^[1]; Qiang, Ji ^[2]; Hao, Yue ^[1]

---

1. Michigan State Univ., East Lansing, MI (United States)
2. Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)

Efficient accelerator modeling and particle tracking are key for the design and configuration of modern particle accelerators. In this work, we present JuTrack, a nested accelerator modeling package developed in the Julia programming language and enhanced with compiler-level automatic differentiation (AD). With the aid of AD, JuTrack enables rapid derivative calculations in accelerator modeling, facilitating sensitivity analyses and optimization tasks. Here we demonstrate the effectiveness of AD-derived derivatives through several practical applications, including sensitivity analysis of space-charge-induced emittance growth, nonlinear beam dynamics analysis for a synchrotron light source, and lattice parameter tuning of the future Electron-Ion Collider (EIC). Through the incorporation of automatic differentiation, this package opens up new possibilities for accelerator physicists in beam physics studies and accelerator design optimization.

• Research Organization: Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)
• Sponsoring Organization: USDOE; USDOE Energy Information Administration (EIA)
• Grant/Contract Number: AC02-05CH11231; SC0024170
• OSTI ID: 2544245
• Journal Information: Computer Physics Communications, Journal Name: Computer Physics Communications Vol. 309; ISSN 0010-4655
• Publisher: ElsevierCopyright Statement
• Country of Publication: United States
• Language: English