torch._utils
torch
collections
torch.nn.modules.container
builtins
typing
itertools
torch.jit
copy
inspect
warnings
re
contextlib
functools
torch._dynamo
torch.autograd
torch.nn
torch.fx
opt_einsum_fx
random
math
fractions
io
pickle
numpy
contextvars
torch.nn.functional
torch.nn.modules.sparse
openequivariance
os
packaging.version
importlib.metadata
cuequivariance
cuequivariance.group_theory.irreps_array.irrep_utils
cuequivariance_torch
cuequivariance.group_theory.experimental.e3nn
torch.nn.modules.activation
torch.fx.experimental.proxy_tensor
difflib
uuid
