pyproject.toml

[tool.poetry]
name = "feature-dynamics"
version = "0.1.0"
description = "Planting language models, seeing how they grow etc."
authors = ["Niels Horn <niels@horn.ninja>"]
license = "MIT license"
readme = "README.md"
packages = [
    { include = "models" },
    { include = "training" }
]

[tool.poetry.dependencies]
python = "~3.11"
torch = "^2.2.0"
einops = "^0.7.0"
pytest = "^8.0.1"
accelerate = "^0.27.2"
datasets = "^2.17.1"
pydantic = "^2.6.1"
syntaxi = {git = "https://github.com/nilq/syntaxi"}
transformers = "^4.38.1"
wandb = "^0.16.3"
black = "^24.2.0"
huggingface-hub = "^0.20.3"
typer = "^0.9.0"
structlog = "^24.1.0"
evaluate = "^0.4.1"
ruff = "^0.2.2"
mergekit = {path = "../mergekit"}
ipdb = "^0.13.13"
scikit-learn = "^1.4.1.post1"
transformer-lens = {git = "https://github.com/nilq/TransformerLens"}
matplotlib = "^3.8.3"
seaborn = "^0.13.2"
bokeh = "^3.4.0"
umap-learn = "^0.5.6"
openai = "^1.23.2"
h5py = "^3.11.0"


[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"