MolecularDiffusion.modules.layers.equiformer_v2.radial_function

Copyright (c) Meta Platforms, Inc. and affiliates.

Classes

RadialFunction

Contruct a radial function (linear layers + layer normalization + SiLU) given a list of

Module Contents

class MolecularDiffusion.modules.layers.equiformer_v2.radial_function.RadialFunction(channels_list)

Bases: torch.nn.Module

Contruct a radial function (linear layers + layer normalization + SiLU) given a list of channels.

forward(inputs)

net