XBTorch API

This section contains the full API reference, generated automatically.

xbtorch

XBTorch root package

xbtorch.decomposition

Gradient decomposition algorithms

xbtorch.deployment

Deployment (mapping, encoding, etc.) of solutions to inference accelerators

xbtorch.devices

Device models and device model presets

xbtorch.loss

Custom loss functions and tools for loss landscapes

xbtorch.nn

Models, datasets, and various XBTorch utilities akin to torch.nn

xbtorch.patches

Decorators for patching PyTorch models and optimizers for XBTorch

xbtorch.quant

Tools for various quantization-related tasks

xbtorch.optim

XBTorch-wrapped optimizers with support for WAGE quantization, device-aware updates, and decomposition algorithms