cuet.TensorProduct
andcuet.EquivariantTensorProduct
now require inputs to be of shape(batch_size, dim)
or(1, dim)
. Inputs of dimension(dim,)
are no more allowed.cuet.TensorProduct
andcuet.EquivariantTensorProduct
are no more variadic functions. They now require a list oftorch.Tensor
as input.cuex.IrrepsArray
is an alias forcuex.RepArray
.cuex.RepArray.irreps
andcuex.RepArray.segments
are not functions anymore. They are now properties.cuex.IrrepsArray.is_simple
is replaced bycuex.RepArray.is_irreps_array
.- The function
cuet.spherical_harmonics
is replaced by the Torch Modulecuet.SphericalHarmonics
. This was done to allow the use oftorch.jit.script
andtorch.compile
.
- Support of
torch.jit.script
andtorch.compile
. Known issue: the export in c++ is not working. - Add
cue.IrrepsAndLayout
: A simple class that inherits fromcue.Rep
and contains acue.Irreps
and acue.IrrepsLayout
. - Add
cuex.RepArray
for representing an array of any kind of representations (not only irreps like before withcuex.IrrepsArray
).
- Add support for empty batch dimension in
cuet
(cuequivariance_torch
).
- Beta version of cuEquivariance released.