#33

refactor causal attention masking to support block masks

Add comprehensive causal and custom masking support

docs: update README.md

Documentation

Mixed precision

Fixed the register pressure issues on M1, and improved performance on M3

C++ reference implementation of GEMM

Replace the main branch with the rewritten implementation

Feature/larger batches