Module: TorchRec::Modules

Defined in:

lib/torchrec/modules/utils.rb,
lib/torchrec/modules/mlp/mlp.rb,
lib/torchrec/modules/mlp/perceptron.rb,
lib/torchrec/modules/activation/swish_layer_norm.rb

Defined Under Namespace

Modules: Activation, MLP, Utils