"""
mlx-fisher: GPU-accelerated Fisher Information Matrix computation on Apple Silicon.

Author: Sheng-Kai Huang <akai@fawstudio.com>
License: MIT
"""

__version__ = "0.1.0"
__author__ = "Sheng-Kai Huang"

from mlx_fisher.fisher import (
    FisherMatrix,
    fisher_matrix_cl,
    kl_divergence,
    quantum_relative_entropy,
)
from mlx_fisher.natural_gradient import NaturalGradientOptimizer

__all__ = [
    "FisherMatrix",
    "fisher_matrix_cl",
    "kl_divergence",
    "quantum_relative_entropy",
    "NaturalGradientOptimizer",
]
