flow.equilibrium

flow.equilibrium(rho, u)

The discrete Maxwellian, expanded to second order in velocity.

Parameters

Name	Type	Description	Default
rho	jax.Array	Density, (NX, NY, NZ).	required
u	jax.Array	Velocity, (3, NX, NY, NZ).	required

Returns

Name	Type	Description
	jax.Array	Equilibrium populations, (19, NX, NY, NZ).