Feature: Multi-Head Attention #540

Job	Run time
code-coverage	3m 14s
run_tests (ubuntu-latest, 3.9)	3m 3s
run_tests_notebooks (ubuntu-latest, 3.9)	5m 1s
run_tests (ubuntu-latest, 3.10)	2m 54s
run_tests_notebooks (ubuntu-latest, 3.10)	5m 3s
run_tests (ubuntu-latest, 3.11)	3m 0s
run_tests_notebooks (ubuntu-latest, 3.11)	4m 59s
run_tests (macos-latest, 3.9)	6m 10s
run_tests_notebooks (macos-latest, 3.9)	7m 0s
run_tests (macos-latest, 3.10)	2m 57s
run_tests_notebooks (macos-latest, 3.10)	10m 55s
run_tests (macos-latest, 3.11)	4m 35s
run_tests_notebooks (macos-latest, 3.11)	5m 44s
	1h 4m 35s

Provide feedback