[project]
name = "needle-export"
version = "0.0.1"
description = "Export Cactus Needle to ONNX via PyTorch port"
requires-python = ">=3.11"
dependencies = [
    "torch",
    "onnx",
    "onnxruntime",
    "jax",
    "jaxlib",
    "flax",
    "sentencepiece",
    "transformers",
    "huggingface_hub",
    "numpy",
    "needle",
    "onnxscript>=0.7.0",
]

[tool.uv.sources]
needle = { path = "../external/needle", editable = true }