JUTRACK: a Julia package for auto-differentiable accelerator modeling and particle tracking

Jinyu Wan,Yue Hao,Helena Alamprese,Christian Ratcliff,Ji Qiang
2024-10-01
Abstract: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 Julia programing 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. 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.
Accelerator Physics
What problem does this paper attempt to address?