IrrepsLinear

class cuequivariance_jax.nnx.IrrepsLinear

__init__(

irreps_in,

irreps_out,

scale=1.0,

*,

precision=None,

dtype=<class 'jax.numpy.float32'>,

rngs,

)

Parameters:

• irreps_in (Irreps)

• irreps_out (Irreps)

• scale (float)

• precision (Precision | None)

• dtype (Any)

• rngs (Rngs)