All modules for which code is available

• symm_learning.linalg
• symm_learning.models.control.cond_transformer
• symm_learning.models.control.econd_transformer
• symm_learning.models.diffusion.cond_eunet1d
• symm_learning.models.diffusion.cond_unet1d
• symm_learning.models.emlp
• symm_learning.models.time_cnn.cnn_encoder
• symm_learning.models.time_cnn.ecnn_encoder
• symm_learning.nn.activation
• symm_learning.nn.conv
• symm_learning.nn.disentangled
• symm_learning.nn.distributions
• symm_learning.nn.linear
• symm_learning.nn.module
• symm_learning.nn.normalization
• symm_learning.nn.parametrizations
• symm_learning.nn.pooling
• symm_learning.nn.running_stats
• symm_learning.nn.transformer.etransformer
• symm_learning.nn.transformer.transformer
• symm_learning.representation_theory
• symm_learning.stats