Package `nux`

Provide high-level Python APIs to access Furiosa AI's NPUs and its eco-system

Expand source code

"""Provide high-level Python APIs to access Furiosa AI's NPUs and its eco-system"""
__version__ = "0.1.2"
__all__ = ["model", "session", "tensor", "quantizer"]

import logging

from nux.errors import NativeError, is_ok, is_err
from . import session, model, tensor, quantizer

logging.basicConfig()
LOG = logging.getLogger(__name__)
LOG.setLevel(logging.INFO)

Sub-modules

nux.errors: Nux Exception and Error
nux.model: Model and its methods to access model metadata
nux.quantizer: Helper Utility to transform onnx models to quantized models …
nux.session: Session and its asynchronous API for model inference
nux.tensor: Tensor object and its utilities