Skip to content
Open
Show file tree
Hide file tree
Changes from 14 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
13142fb
init: Added some stuff, tons to go
harshaljanjani Sep 16, 2025
e4493b4
Merge branch 'keras-team:master' into gemma3n
harshaljanjani Oct 11, 2025
c0bcfc4
refactor: Update according to changed MobileNetV5 design
harshaljanjani Oct 11, 2025
cb3e1b0
fix: Reduce memory usage in tests to prevent OOM errors
harshaljanjani Oct 11, 2025
9849a7c
feat: Add checkpoint conversion script (1 preset)
harshaljanjani Oct 31, 2025
c2cb963
init: task model with tests
harshaljanjani Nov 4, 2025
e3c257d
exports: I like the CI green.
harshaljanjani Nov 4, 2025
c8fc40f
skipif: Keras version compatibility in CausalLM tests
harshaljanjani Nov 4, 2025
ddd29bb
refactor: Gemma3nCausalLM and update tests
harshaljanjani Nov 5, 2025
093bff8
feat: Complete TODO for CausalLM
harshaljanjani Nov 7, 2025
7aa2d5e
nit: Minor changes
harshaljanjani Nov 8, 2025
6d288e8
fix: Lower the divergence to single digits.
harshaljanjani Dec 17, 2025
11fe7b6
Merge branch 'keras-team:master' into gemma3n
harshaljanjani Dec 25, 2025
5709445
Implement kv sharing, causal masking, fix other causes of numerical …
laxmareddyp Feb 19, 2026
417a359
Merge branch 'keras-team:master' into gemma3n_model
laxmareddyp Feb 20, 2026
95c903f
Fix Gemma3n parameter count mismatch and logits divergence
laxmareddyp Feb 20, 2026
f4b0a71
Fix Gemma3n CI: handle 4D attention mask in decoder and update param…
laxmareddyp Feb 23, 2026
65fc602
Format Fix
laxmareddyp Feb 23, 2026
729f0af
Fix Gemma3n generate: squeeze 4D attention mask to 2D in decoder
laxmareddyp Feb 23, 2026
537e0ad
Revert the MobileNetV5 changes to make backward compatability with t…
laxmareddyp Feb 24, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions keras_hub/api/layers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,12 @@
from keras_hub.src.models.gemma3.gemma3_image_converter import (
Gemma3ImageConverter as Gemma3ImageConverter,
)
from keras_hub.src.models.gemma3n.gemma3n_audio_converter import (
Gemma3nAudioConverter as Gemma3nAudioConverter,
)
from keras_hub.src.models.gemma3n.gemma3n_image_converter import (
Gemma3nImageConverter as Gemma3nImageConverter,
)
from keras_hub.src.models.hgnetv2.hgnetv2_image_converter import (
HGNetV2ImageConverter as HGNetV2ImageConverter,
)
Expand Down
12 changes: 12 additions & 0 deletions keras_hub/api/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,6 +315,18 @@
from keras_hub.src.models.gemma3.gemma3_vision_encoder import (
Gemma3VisionEncoder as Gemma3VisionEncoder,
)
from keras_hub.src.models.gemma3n.gemma3n_backbone import (
Gemma3nBackbone as Gemma3nBackbone,
)
from keras_hub.src.models.gemma3n.gemma3n_causal_lm import (
Gemma3nCausalLM as Gemma3nCausalLM,
)
from keras_hub.src.models.gemma3n.gemma3n_causal_lm_preprocessor import (
Gemma3nCausalLMPreprocessor as Gemma3nCausalLMPreprocessor,
)
from keras_hub.src.models.gemma3n.gemma3n_tokenizer import (
Gemma3nTokenizer as Gemma3nTokenizer,
)
from keras_hub.src.models.gpt2.gpt2_backbone import GPT2Backbone as GPT2Backbone
from keras_hub.src.models.gpt2.gpt2_causal_lm import (
GPT2CausalLM as GPT2CausalLM,
Expand Down
3 changes: 3 additions & 0 deletions keras_hub/api/tokenizers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,9 @@
from keras_hub.src.models.gemma3.gemma3_tokenizer import (
Gemma3Tokenizer as Gemma3Tokenizer,
)
from keras_hub.src.models.gemma3n.gemma3n_tokenizer import (
Gemma3nTokenizer as Gemma3nTokenizer,
)
from keras_hub.src.models.gpt2.gpt2_tokenizer import (
GPT2Tokenizer as GPT2Tokenizer,
)
Expand Down
Empty file.
Loading
Loading