Add mimo-audio model - #132

Open

Mozoltov821 wants to merge 23 commits into

jax-ml:mainfrom

Mozoltov821:feat-support-mimo-audio

Commits on Jan 19, 2026

Qwen2 support
Mozoltov821
committed
2b31681
doc(qwen2): add readme
Mozoltov821
committed
25bec11
test: qwen2
Mozoltov821
committed
570a469
mimo-audio support
Mozoltov821
committed
8345bd1
mimo-audio-tokenizer support
Mozoltov821
committed
c8bc92e
add mimo-audio readme
Mozoltov821
committed
a64b513
test: mimo-audio
Mozoltov821
committed
f0cc7a0
test: qwen2
Mozoltov821
committed
f67af1e
test: mimo-audio
Mozoltov821
committed
ff21293

Commits on Jan 20, 2026

fix: audio encoder and audio decoder padding config
Mozoltov821
committed
f65d52e
test: mimo audio
Mozoltov821
committed
fb4e252
test: mimo_audio_tokenizer
Mozoltov821
committed
710ce6d
pre-commit fixes
Mozoltov821
committed
27c7553

Commits on Jan 22, 2026

Merge branch 'main' into feat-support-mimo-audio
chapman20j
authored
4034f85
Merge branch 'main' into feat-support-mimo-audio
jenriver
authored
dcd7879
Merge branch 'main' into feat-support-mimo-audio
jenriver
authored
668ec1c

Commits on Feb 3, 2026

Removing dinov3 model output class to facilitate jit compilation (#130)

authored and
Mozoltov821
committed
200d217
Fixed convnext tests and nnx warnings (#135)

authored and
Mozoltov821
committed
c9ecb68
fix: correct checkbox syntax in pull request template (#139)

Moriyuki-S
authored and
Mozoltov821
committed
1f41221
Mamba2: SSM State Caching for Efficient Incremental Generation (#131)

authored and
Mozoltov821
committed
4978280
Temporarily disable Gemini agents until specific review guidelines are configured. (#140)

jenriver
authored and
Mozoltov821
committed
077f43b
Merge remote-tracking branch 'origin/feat-support-mimo-audio' into feat-support-mimo-audio
Mozoltov821
committed
0cfdcbe
Merge branch 'main' into feat-support-mimo-audio
jenriver
authored
6b354ea