Add decoder head modules

[mojtababahrami]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 75.09804% with 127 lines in your changes missing coverage. Please review.
✅ Project coverage is 65.97%. Comparing base (3ba6a5e) to head (f9d7c3c).
⚠️ Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
src/concept/decoder/train_decoder.py	69.04%	39 Missing ⚠️
src/concept/api.py	56.47%	37 Missing ⚠️
src/concept/decoder/decoder_model.py	84.02%	31 Missing ⚠️
src/concept/modules/flash_attention_layer.py	75.00%	17 Missing ⚠️
src/concept/modules/mha.py	86.36%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #19      +/-   ##
==========================================
+ Coverage   64.17%   65.97%   +1.79%     
==========================================
  Files          19       22       +3     
  Lines        2434     2930     +496     
==========================================
+ Hits         1562     1933     +371     
- Misses        872      997     +125     

Files with missing lines	Coverage Δ
src/concept/decoder/__init__.py	100.00% <100.00%> (ø)
src/concept/modules/transformer.py	97.14% <100.00%> (+1.68%)	⬆️
src/concept/modules/mha.py	84.00% <86.36%> (+1.74%)	⬆️
src/concept/modules/flash_attention_layer.py	80.76% <75.00%> (-6.33%)	⬇️
src/concept/decoder/decoder_model.py	84.02% <84.02%> (ø)
src/concept/api.py	71.22% <56.47%> (-4.15%)	⬇️
src/concept/decoder/train_decoder.py	69.04% <69.04%> (ø)

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.