Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
GNU General Public License v3.0
33
stars
13
forks
source link
Replace custom histogramdd() with torch.histogramdd() #18
Closed
jank324 closed 1 year ago
Fixes #10, simplifying the code and giving a ca. 5x speedup when using screen readings.