Compute matrix trace and support updates | Anthropic Interview Question