fix(ref): widen qwen2 week 2 offset type hints

[Connor1996]

Summary

• update the Qwen2 week 2 reference signatures to match the batched offsets already supported at runtime
• allow scalar offsets, Python lists, and mx.array values consistently across model, block, and attention entry points

Testing

• not run (per request)