Minimal implementation of QED synchrotron extension for PIC codes as described in: https://journals.aps.org/pre/abstract/10.1103/PhysRevE.92.023305
Please install from requirements.txt as well as this library: https://github.com/sbarratt/torch_interpolations
On mobile NVidia P3200 current setup: ~60ms per iteration