e3nn/e3nn:

[0.2.4] - 2021-03-23 Added

• basis='smooth_finite' option to math.soft_one_hot_linspace
• math.soft_unit_step function
• nn.model.v2103 generic message passing model + examples of networks using it.
• o3.TensorProduct: is jit scriptable
• o3.TensorProduct: also broadcast the weight argument
• simple e3nn models can be saved/loaded with torch.save()/torch.load()
• JITable o3.SphericalHarmonics module version of o3.spherical_harmonics
• in_place option for e3nn.util.jit compilation functions
• New @compile_mode("unsupported") for modules that do not support TorchScript
• flake8 settings have been added to setup.cfg for improved code style
• TensorProduct.visualize() can now plot weights
• basis='bessel' option to math.soft_one_hot_linspace

• o3.TensorProduct now uses torch.fx to generate it's code
• e3nn now requires the latest stable PyTorch, >=1.8.0
• in soft_one_hot_linspace the argument base is renamed into basis
• Irreps.slices(), do zip(irreps.slices(), irreps) to retrieve the old behavior
• math.soft_one_hot_linspace very small change in the normalization of fourier basis
• normalize2mom is now a torch.nn.Module
• rename arguments set_ir_... into filter_ir_...
• Renamed e3nn.nn.Gate argument irreps_nonscalars to irreps_gated
• Renamed e3nn.o3.TensorProduct arguments x1, x2 to x, y

• nn.Gate was crashing when the number of scalars or gates was zero
• device edge cases for Gate and SphericalHarmonics