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

TL;DR

JuTrack is a Julia-based accelerator modeling package that leverages automatic differentiation to enable rapid sensitivity analysis and optimization, improving accelerator design and particle tracking efficiency.

Contribution

The paper introduces JuTrack, a novel Julia package that integrates automatic differentiation for accelerator modeling and particle tracking, enhancing analysis speed and accuracy.

Findings

Effective sensitivity analysis of space-charge effects

Improved nonlinear beam dynamics analysis

Facilitated lattice parameter tuning for EIC

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 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. 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.