Package ai.djl.nn.transformer

Contains blocks for transformer models.

See: Description

Class Summary

Class	Description
ScaledDotProductAttentionBlock	A Block implementing scaled product attention according to Vaswani et.
ScaledDotProductAttentionBlock.Builder	A builder for ScaledDotProductAttentionBlocks.
TransformerBaseBlock	Utility base class for implementing custom blocks.

Package ai.djl.nn.transformer Description

Contains blocks for transformer models.