Chenran Xu

Hi, I noticed that there is no requirements file in the repo. Several packages are needed to run the sample. It would be nice if one can just install all...

Inconsistency in the `Generator.generate()` returns

2

Now the the returned values from `generate(n_candidates)` for different Generators are in different conventions. Specifically, some return `List[Dict[str, float]]` and some return `List[Dict[str, np.ndarray]]`, which is not documented explicitly in...

Need more distribution options for bunch generation

6

Currently Cheetah only supports generation of Gaussian bunches. More distributions should be supported, especially allowing to specify different distributions in transversal and longitudinal plane. For example - Truncated gaussian bunch...

What is the purpose of Beam.empty?

3

In the `ParticleBeam`, there are some locations where it's checked whether `self is not Beam.empty`, e.g. https://github.com/desy-ml/cheetah/blob/b92502966f01797889697f1dcba46a2667328f8c/cheetah/particles.py#L1023 But this is never the case, since one cannot initialize the ParticleBeam without...

Add proper documentation for the converters

1

Right now we support converters for 3 different simulation codes (OCELOT, ASTRA, and Bmad), and each behaves differently. Documentation of what each converter does is needed. One can either -...

Implement more lattice elements for cheetah

2

Many lattice element types are still missing in Cheetah, and I'm working on adding these. - [x] Bending magnets - [x] Sector Bend - [x] Rect Bend - [x] Skew...

## Description - Add batch dimensions for the rotation matrix, now the batched tracking works properly with tilted elements - Add batch dimensions for the misalignment matrix. Now the batched...

Missing batch-execution compatibility in Dipole, Rotation matrix, etc.

4

Rotation matrix is currently not batch-compatible. This is causing problem when elements are rotated (tilt!=0) ```python import torch from cheetah import ParticleBeam, Drift, Segment, Quadrupole beam_in = ParticleBeam.from_parameters(num_particles=torch.tensor(1000), energy=torch.tensor([1e9,1e9]), mu_x=torch.tensor([1e-5,...

## Description ### Rename coordinate `s` to `tau` To avoid confusion and make space for the variable `s` denoting the position of reference particle along a beamline, the 5-th component...

As discussed before and to prevent confusion for users, the `ParticleBeam.particles` should only return the 6d phase space coordinates, and the 7d structure can be stored internally e.g. in `particles_7d`