NanoCode012 97f1b1758d Feat: add kimi linear support (#3257) ...

* feat: add custom kimi linear patch [skip ci]

* feat: add configuration file and fix import [skip ci]

* fix: hijack tokenizer temporarily [skip ci]

* chore: remove accidental commit

* fix: attempt patch kimi remote

* fix: kwargs passsed

* fix: device for tensor

* fix: aux loss calculation

* feat: cleaned up patches order

* fix: remove duplicate tokenizer patch

* chore: add debug logs

* chore: add debug logs

* chore: debug

* Revert "chore: add debug logs"

This reverts commit da372a5f67.

* Revert "chore: add debug logs"

This reverts commit 97d1de1d7c.

* fix: KeyError: 'tokenization_kimi'

* fix: support remote_model_id in cce patch

* feat: add config preload patch

* fix: use standard aux loss calc and updated modeling

* fix: import

* feat: add kimi-linear docs and example

* chore: add note about moe kernels

* feat: update cce to include kimi-linear

* chore: lint

* chore: update main readme

* fix: patch mechanism to address comments

* chore: lint

* fix: tests

* chore: cleanup comment

2025-12-25 17:53:52 +07:00

..

alst

feat(doc): add optimizations table of content to our improvements (#3175) [skip ci]

2025-09-24 16:13:49 -04:00

apertus

feat: add apertus model and cce (#3144) [skip ci]

2025-09-19 17:34:04 +07:00

arcee

feat: add apertus model and cce (#3144) [skip ci]

2025-09-19 17:34:04 +07:00

archived

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

cloud

deploy training jobs to baseten w truss in axolotl cli (#3086) [skip ci]

2025-08-26 09:29:50 -04:00

cohere

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

colab-notebooks

Feat: add kimi linear support (#3257)

2025-12-25 17:53:52 +07:00

deepcogito

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

deepseek-v2

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

devstral

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00

distributed-parallel

feat: update nd parallelism readme (#3039)

2025-08-08 12:45:36 +01:00

falcon-h1

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

gemma2

make pad_to_sequence_len default to the same value as sample_packing (#2941) [skip ci]

2025-07-21 11:40:56 -04:00

gemma3

fix(example): set model_type to load for gemma3 text (#3242)

2025-11-04 07:35:07 +07:00

gemma3n

Feat: add Magistral Small 2509 and native mistral3 tokenizer support (#3165)

2025-09-18 15:42:20 +07:00

glm4

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

gpt-oss

feat: add trackio as experiment tracking integration (#3253)

2025-12-23 08:49:07 -05:00

granite4

feat: add granite4 examples (#3256) [skip ci]

2025-11-13 10:19:16 -05:00

hunyuan

Feat: add hunyuan v1 (#3016)

2025-09-10 09:03:30 +07:00

jamba

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

kimi-linear

Feat: add kimi linear support (#3257)

2025-12-25 17:53:52 +07:00

LiquidAI

feat: add lfm2 family and latest moe model (#3208)

2025-10-09 10:47:41 -04:00

llama-2

Add FSDP v2 swap memory support + QLoRA compatibility fixes (#3167)

2025-09-26 10:23:59 +01:00

llama-3

fix check for fp8 capability (#3324)

2025-12-22 13:58:25 -05:00

llama-3-vision

fix: revert changing default optimizer to muon (#2965) [skip ci]

2025-07-22 10:00:30 -04:00

llama-4

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

llava

fix: revert changing default optimizer to muon (#2965) [skip ci]

2025-07-22 10:00:30 -04:00

magistral

Feat: add ministral3 (#3297)

2025-12-04 08:32:08 -05:00

mamba

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

ministral

fix: improve ministral3 docs to be clearer (#3300)

2025-12-04 21:44:44 +07:00

ministral3

fix: improve ministral3 docs to be clearer (#3300)

2025-12-04 21:44:44 +07:00

mistral

fix: force train split for json,csv,txt for test_datasets and misc doc changes (#3226)

2025-10-22 15:23:20 -07:00

olmo3

Feat: add ministral3 (#3297)

2025-12-04 08:32:08 -05:00

orpheus

use warmup_ratio as a better default than warmup steps since it's data dependent (#2897) [skip ci]

2025-07-30 06:44:06 -04:00

phi

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

pixtral

Feat: add Magistral Small 2509 and native mistral3 tokenizer support (#3165)

2025-09-18 15:42:20 +07:00

qat_nvfp4

Add QAT NVFP4 configs for blogpost (#3280) [skip ci]

2025-12-17 09:35:22 -05:00

qwen2

Distributed Muon Optimizer (#3264)

2025-12-19 10:43:47 -05:00

qwen2_5-vl

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

qwen2-vl

fix: unify default for conversations_field [skip-e2e] (#3070)

2025-09-23 21:22:15 +07:00

qwen3

fix: update qwen3 jinja tokenization off a few tokens (#3295)

2025-12-09 14:31:03 +07:00

qwen3-next

feat(qwen3-next): Adds targeting of shared expert and attention modules (#3183)

2025-09-25 17:06:16 +07:00

seed-oss

Feat: add Olmo3 (BC with Olmo and Olmo2) (#3275)

2025-11-24 10:21:31 +07:00

slurm

Example for Slurm and various fixes (#3038) [skip ci]

2025-08-08 08:02:03 -04:00

smolvlm2

Feat: add Olmo3 (BC with Olmo and Olmo2) (#3275)

2025-11-24 10:21:31 +07:00

streaming

Streaming SFT support (#3101)

2025-09-02 12:08:44 -04:00

trinity

Feat: add trinity by ArceeAI (#3292)

2025-12-02 13:12:55 -05:00

voxtral

fix: voxtralprocessor broken (#3255) [skip ci]

2025-11-13 10:18:42 -05:00