import numpy as np


def causal_mask(T):
    raise NotImplementedError


def scaled_dot_product_attention(Q, K, V, mask=None):
    raise NotImplementedError