| [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 } | |