Skip to main content

hydrogym.jax.envs.channel

ChannelEnvParams Objects​

@struct.dataclass
class ChannelEnvParams(BaseEnvParams)

Extends base EnvParams with channel-specific settings.

nsteps​

DNS substeps per RL step

SpectralState Objects​

class SpectralState(NamedTuple)

u_hat​

(Nx,Ny,Nz), complex

ChannelFlowSpectralEnv Objects​

class ChannelFlowSpectralEnv(JAXFlowEnvBase)

3D turbulent channel flow environment using a pseudo-spectral DNS solver.