"""GPU-accelerated tensor operations for FusionAGI.

Auto-selects the best available backend:
- TensorFlow with TensorCore/mixed-precision (when installed)
- NumPy CPU fallback (always available)

Install GPU support: pip install fusionagi[gpu]
"""

from fusionagi.gpu.backend import (
    DeviceType,
    NumPyBackend,
    TensorBackend,
    get_backend,
    reset_backend,
)
from fusionagi.gpu.tensor_attention import (
    attention_consensus,
    cross_claim_attention,
)
from fusionagi.gpu.tensor_scoring import (
    gpu_score_claims_against_reference,
    gpu_score_hypotheses,
)
from fusionagi.gpu.tensor_similarity import (
    deduplicate_claims,
    nearest_neighbors,
    pairwise_text_similarity,
)
from fusionagi.gpu.training import (
    TrainingConfig,
    TrainingResult,
    optimize_heuristic_weights,
    prepare_training_pairs,
    run_gpu_training,
)

__all__ = [
    "DeviceType",
    "NumPyBackend",
    "TensorBackend",
    "get_backend",
    "reset_backend",
    "deduplicate_claims",
    "nearest_neighbors",
    "pairwise_text_similarity",
    "attention_consensus",
    "cross_claim_attention",
    "gpu_score_claims_against_reference",
    "gpu_score_hypotheses",
    "TrainingConfig",
    "TrainingResult",
    "optimize_heuristic_weights",
    "prepare_training_pairs",
    "run_gpu_training",
]