All modules for which code is available

• e3tools._pack_unpack
• e3tools._radius
• e3tools._scatter
• e3tools.nn._axis_to_mul
• e3tools.nn._conv
• e3tools.nn._extract_irreps
• e3tools.nn._gate
• e3tools.nn._interaction
• e3tools.nn._layer_norm
• e3tools.nn._mlp
• e3tools.nn._scaling
• e3tools.nn._tensor_product
• e3tools.nn._transformer