| .. |
|
__init__.py
|
Mlx llm package (#301)
|
2024-01-12 10:25:56 -08:00 |
|
base.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
cache.py
|
support different k and v head dims
|
2024-10-31 16:28:33 -07:00 |
|
cohere.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
dbrx.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
deepseek_v2.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
deepseek.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
gemma2.py
|
override dtype with quant (#1062)
|
2024-10-22 09:56:45 -07:00 |
|
gemma.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
gpt2.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
gpt_bigcode.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
gpt_neox.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
internlm2.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
llama.py
|
fix sed
|
2024-10-31 12:06:34 -07:00 |
|
mamba.py
|
fix mamba models conversion (#1065)
|
2024-10-22 15:44:08 -07:00 |
|
minicpm.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
mixtral.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
nemotron.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
olmo.py
|
More cache improvements (#1015)
|
2024-10-07 20:45:51 -07:00 |
|
openelm.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
phi3.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
phi3small.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
phi.py
|
fix tests
|
2024-10-31 12:37:15 -07:00 |
|
phimoe.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
phixtral.py
|
fix tests
|
2024-10-31 12:37:15 -07:00 |
|
plamo.py
|
fix tests
|
2024-10-31 12:37:15 -07:00 |
|
qwen2_moe.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
qwen2.py
|
fix sed
|
2024-10-31 12:06:34 -07:00 |
|
qwen.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
recurrent_gemma.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
stablelm.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
starcoder2.py
|
single sdpa function
|
2024-10-31 12:05:27 -07:00 |
|
su_rope.py
|
Add Phi-3.5-MoE (#946)
|
2024-08-24 06:52:33 -07:00 |
|
switch_layers.py
|
Handle longer prompt/generation (#931)
|
2024-08-16 15:28:39 -07:00 |
