Apple Interview Question

Implement masked multi-head self-attention for a transformer from scratch