eModule

class eModule(*args, **kwargs)

Bases: Module

Lightweight base class for equivariant modules.

This base centralizes lifecycle behavior related to cache validity and mode transitions. Subclasses are expected to: - define in_rep and out_rep when requires_reps=True - optionally override invalidate_cache

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Parameters:

• args (Any)

• kwargs (Any)

invalidate_cache()

Clear derived cached tensors so they are recomputed on next use.

Return type:

None