[project]
name = "gemma4-jax"
version = "0.1.0"
description = "From-scratch Flax NNX port of google/gemma-4-12B (full multimodal: text + encoder-free vision + audio)"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "jax>=0.4.30",
    "flax>=0.9.0",
]

[project.optional-dependencies]
# needed only to load real weights / tokenize
weights = ["safetensors", "huggingface_hub", "tokenizers"]

[tool.setuptools.packages.find]
include = ["gemma4_jax*"]