57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
|
"""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",
|
||
|
|
]
|