All modules for which code is available

• nami.components.activation
• nami.components.mlp
• nami.components.time
• nami.components.transformer
• nami.core.broadcast
• nami.core.specs
• nami.core.typing
• nami.debug
• nami.diagnostics
• nami.distributions.base
• nami.distributions.normal
• nami.divergence.base
• nami.divergence.exact
• nami.divergence.hutchinson
• nami.fields.base
• nami.fields.diffusion
• nami.fields.transformer
• nami.fields.velocity
• nami.interpolants.gamma
• nami.interpolants.transforms
• nami.lazy
• nami.losses.fm
• nami.losses.stochastic_fm
• nami.masking
• nami.paths.base
• nami.paths.cosine
• nami.paths.linear
• nami.processes.diffusion
• nami.processes.fm
• nami.schedules.base
• nami.schedules.edm
• nami.schedules.ve
• nami.schedules.vp
• nami.solvers.heun
• nami.solvers.ode
• nami.solvers.sde
• nami_toys.checkerboard
• nami_toys.dataset
• nami_toys.gaussian
• nami_toys.moons
• nami_toys.parameterised
• nami_toys.ring
• nami_toys.rng
• nami_toys.shell
• nami_toys.spirals
• nami_toys.standardise