Development Guide
Environment
python -m venv .venv
source .venv/bin/activate # Windows: .venv\Scripts\activate
pip install -r requirements.txt
pip install -e .
Code style & checks
- Format: Black
- Lint: flake8
- Types: mypy
- Tests: pytest
black src/ tests/
flake8 src/ tests/
mypy src/
pytest -q
Project layout
train_ultrathink.py— entry point for trainingsrc/models/— model architecture and componentssrc/data/— datasets and loaderssrc/training/— distributed, RLHF utilitiessrc/evaluation/— benchmarks
Making changes
- Create a branch
- Add tests for new behavior
- Update docs if user-facing changes
- Run checks before opening a PR