pydantic<3,>=2.0
pyyaml<7,>=6.0
numpy<3,>=1.24
defusedxml<1,>=0.7.1

[arrow]
pyarrow>=15.0

[cpp]
pybind11>=2.12.0
scikit-build-core>=0.9.0

[cuda]
torch>=2.2.0

[dev]
pytest>=7.0
pytest-cov>=4.0
pytest-asyncio>=0.23.0
pytest-timeout>=2.3.0
jsonschema>=4.0
mypy==2.3.1
types-PyYAML>=6.0
pre-commit>=3.0
black==26.5.1
ruff==0.16.4
bandit[toml]==1.9.4
hypothesis>=6.0
pytest-xdist>=3.5
import-linter>=2.0
schemathesis>=4.24
diff-cover>=9.0
pyrefly>=1.2.0
zizmor>=1.29.0
deptry>=0.20

[encryption]
cryptography>=42.0

[experiment]
scipy>=1.11
matplotlib>=3.7

[gcs]
google-cloud-storage>=2.14

[grpc]
grpcio>=1.60.0
grpcio-tools>=1.60.0
protobuf>=4.25.0

[jax]
jax>=0.4.28
jaxlib>=0.4.28
flax>=0.8.2
optax>=0.2.2
orbax-checkpoint>=0.5.0

[notebook]
ipython>=8.0

[otel]
opentelemetry-api>=1.20
opentelemetry-sdk>=1.20
opentelemetry-exporter-otlp-proto-grpc>=1.20

[s3]
boto3>=1.34

[safetensors]
safetensors>=0.4.3

[smt]
z3-solver>=4.12
cvc5>=1.1

[system]
psutil>=5.9

[triton]
triton>=2.1.0
