All modules for which code is available

• collections
• jaxtyping
• optax.schedules._schedule
• ptyrax.dataset
• ptyrax.experiment
• ptyrax.field
• ptyrax.initializers
• ptyrax.logger
• ptyrax.models.detector
• ptyrax.models.illumination
• ptyrax.models.interaction
• ptyrax.models.propagation
• ptyrax.models.ptychography
• ptyrax.parametrizations
• ptyrax.reconstruct
• ptyrax.simulate
• ptyrax.spatial
• ptyrax.training
• ptyrax.utils