mlx/python
Awni Hannun 3c3e558c60
Support transposed head/seq for kv (#1950)
* support transposed head/seq for kv

* fix flaky test

* nit
2025-03-10 10:53:45 -07:00
..
mlx Add type hint for _extra_repr (#1948) 2025-03-10 06:05:36 -07:00
src Fix typo in randint docstring (#1932) 2025-03-05 21:48:00 -08:00
tests Support transposed head/seq for kv (#1950) 2025-03-10 10:53:45 -07:00