| # PsiLM on Gemma 4 12B -- Apple Silicon (MLX). Verified 2026-09-06 on macOS / Python 3.11 | |
| # with: mlx 0.32.2, mlx-lm 0.31.3, transformers 5.16.1, torch 2.13.0, huggingface_hub 1.29.0. | |
| # | |
| # pip install -r requirements.txt | |
| # python psilm_infer.py | |
| # | |
| mlx>=0.32 | |
| mlx-lm==0.31.3 # exact: the Gemma 4 loader (psilm.mlx.gemma_loader) goes through | |
| # mlx_lm.utils internals; other versions are untested | |
| transformers>=4.51 # chat template only (apply_chat_template); the model runs in MLX | |
| torch>=2.4 # imported by psilm.stage2.qa (batching helpers); CPU wheel is fine | |
| huggingface_hub>=0.30 # downloads the backbone from mlx-community | |
| numpy>=1.26 | |
| safetensors>=0.4 | |
| einops>=0.8 # --task 2d only (imported by the DPOT-Tiny definition) | |
| # The PsiLM package itself (bridges, staged forward, Gemma loader, FNO, QA builder). | |
| # Alternative to this line: clone the repository and set PSILM_REPO=/path/to/PsiLM. | |
| psilm @ git+https://github.com/ryoji-info/PsiLM | |