MLPBlock

Multilayer perceptron block composed of multiple FeedForward layers.

Abstract Signature:

MLPBlock(in_features: int, hidden_features: int, out_features: int)

PyTorchKerasPaxML / Praxis

PyTorch

API: torchvision.ops.MLP

Strategy: Direct Mapping

Official Docs ↗

Keras

API: keras.Sequential

Strategy: Plugin (keras_sequential_pack)

PaxML / Praxis

API: paxml.layers.MLPBlock

Strategy: Direct Mapping

Official Docs ↗