Source code for geometric_algebra_attention.jax.VectorAttention


from .. import base
from .internal import AttentionBase



[docs]
class VectorAttention(AttentionBase, base.VectorAttention):
    __doc__ = base.VectorAttention.__doc__