mirror of
https://github.com/cactus-compute/needle.git
synced 2026-10-20 06:00:31 +08:00
34 lines
554 B
TOML
34 lines
554 B
TOML
[project]
|
|
name = "needle"
|
|
version = "0.1.0"
|
|
requires-python = ">=3.11"
|
|
dependencies = [
|
|
"jax",
|
|
"jaxlib",
|
|
"flax",
|
|
"optax",
|
|
"datasets",
|
|
"huggingface_hub",
|
|
"transformers",
|
|
"wandb",
|
|
"sentencepiece",
|
|
"scipy",
|
|
"gcsfs",
|
|
"tqdm",
|
|
"google-genai",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
tpu = ["jax[tpu]"]
|
|
gpu = ["jax[cuda12]"]
|
|
|
|
[project.scripts]
|
|
needle = "needle.cli:main"
|
|
|
|
[tool.setuptools.packages.find]
|
|
include = ["needle*"]
|
|
|
|
[build-system]
|
|
requires = ["setuptools>=68.0"]
|
|
build-backend = "setuptools.build_meta"
|