mlx-lm

__init__.py

Mlx llm package (#301 )

2024-01-12 10:25:56 -08:00

afm7.py

Implementation of AFM in MLX (#232 )

2025-06-12 12:24:45 -07:00

baichuan_m1.py

Update baichuan_m1.py (#127 )

2025-04-26 06:30:59 -07:00

base.py

fix quant sdpa and gemma3 (#121 )

2025-04-24 17:15:10 -07:00

bitlinear_layers.py

Feat: add falcon-e support for bitnet models (#268 )

2025-07-07 08:43:52 -07:00

bitnet.py

Feat: add falcon-e support for bitnet models (#268 )

2025-07-07 08:43:52 -07:00

cache.py

fix cast predicate (#243 )

2025-06-17 15:07:31 -07:00

cohere2.py

Updates for causal mask (#40 )

2025-03-21 08:50:44 -07:00

cohere.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

dbrx.py

allow models to be pickled + test (#261 )

2025-06-29 11:27:40 -07:00

deepseek_v2.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

deepseek_v3.py

allow models to be pickled + test (#261 )

2025-06-29 11:27:40 -07:00

deepseek.py

allow models to be pickled + test (#261 )

2025-06-29 11:27:40 -07:00

dots1.py

Adding support for rednote-hilab/dots.llm1.inst (#211 )

2025-07-02 13:03:26 -07:00

ernie4_5_moe.py

Adding ernie4.5 moe (#267 )

2025-07-04 17:13:13 -07:00

ernie4_5.py

Add Ernie4.5 (#263 )

2025-07-02 07:42:38 -07:00

exaone.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

gemma2.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

gemma3_text.py

Add input_embeddings input to generate_step, Gemma 3, Qwen 2 (#179 )

2025-05-14 11:54:51 -07:00

gemma3.py

Add input_embeddings input to generate_step, Gemma 3, Qwen 2 (#179 )

2025-05-14 11:54:51 -07:00

gemma3n.py

Gemma3n text only support (#258 )

2025-06-30 20:03:00 -07:00

gemma.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

glm4.py

Fix GLM4 Repetition (#105 )

2025-04-19 06:53:37 -07:00

gpt2.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

gpt_bigcode.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

gpt_neox.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

granite.py

Add IBM granite model (#1265 )

2025-02-08 15:46:15 -08:00

helium.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

hunyuan.py

fix hunyuan (#286 )

2025-07-09 12:44:35 -07:00

internlm2.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

internlm3.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

kimi_vl.py

fix cast predicate (#243 )

2025-06-17 15:07:31 -07:00

llama4.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

llama.py

Pixtral text support, pipe input_embeddings through llama arch (#181 )

2025-05-16 19:15:00 -07:00

mamba.py

Optimizations for mamba1 (#1213 )

2025-02-03 13:36:08 -08:00

mimo.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

minicpm3.py

feat: add MiniCPM4 model structure code minicpm4.py and minicpm4 mode… (#212 )

2025-06-09 07:58:21 -07:00

minicpm.py

feat: add MiniCPM4 model structure code minicpm4.py and minicpm4 mode… (#212 )

2025-06-09 07:58:21 -07:00

mistral3.py

Pipe input_embeddings through mistral3 model_type (#254 )

2025-06-25 07:16:07 -07:00

mixtral.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

nemotron-nas.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

nemotron.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

olmo2.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

olmo.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

olmoe.py

adding OLMoE architecture (#1321 )

2025-03-05 13:46:06 -08:00

openelm.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

phi3.py

feat: add MiniCPM4 model structure code minicpm4.py and minicpm4 mode… (#212 )

2025-06-09 07:58:21 -07:00

phi3small.py

Updates for causal mask (#40 )

2025-03-21 08:50:44 -07:00

phi.py

allow models to be pickled + test (#261 )

2025-06-29 11:27:40 -07:00

phimoe.py

feat: add MiniCPM4 model structure code minicpm4.py and minicpm4 mode… (#212 )

2025-06-09 07:58:21 -07:00

phixtral.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

pixtral.py

Pixtral text support, pipe input_embeddings through llama arch (#181 )

2025-05-16 19:15:00 -07:00

plamo2.py

Update logits to new mlx + version bump (#63 )

2025-04-03 16:59:12 -07:00

plamo.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

qwen2_moe.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

qwen2.py

Add input_embeddings input to generate_step, Gemma 3, Qwen 2 (#179 )

2025-05-14 11:54:51 -07:00

qwen3_moe.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

qwen3.py

Add Qwen3 and Qwen3-MoE (#41 )

2025-04-28 06:44:17 -07:00

qwen.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

recurrent_gemma.py

nits for qwen3_moe file (#199 )

2025-05-28 21:46:59 -07:00

rope_utils.py

feat: add MiniCPM4 model structure code minicpm4.py and minicpm4 mode… (#212 )

2025-06-09 07:58:21 -07:00

smollm3.py

Add SmolLM3. (#272 )

2025-07-07 08:20:04 -07:00

stablelm.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

starcoder2.py

Length masking for batch inputs (#1173 )

2024-12-18 19:43:52 -08:00

switch_layers.py

Fix MoE fine tuning (#288 )

2025-07-09 12:42:35 -07:00