| [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*"] | |