API Reference

aihwkit_lightning	Analog hardware library for PyTorch.
aihwkit_lightning.exceptions	Custom Exceptions for aihwkit.
aihwkit_lightning.version	Package version string.
aihwkit_lightning.inference	High level inference tools.
aihwkit_lightning.inference.calibration	High level inference tools.
aihwkit_lightning.inference.calibration.calibration	Calibration for inference.
aihwkit_lightning.nn	Neural network module.
aihwkit_lightning.nn.conversion	Digital/analog model conversion utilities.
aihwkit_lightning.optim	Analog Optimizers.
aihwkit_lightning.optim.analog_optimizer	Analog-aware inference optimizer.
aihwkit_lightning.simulator	Simulator module comprising parameters and configs.
aihwkit_lightning.simulator.configs	Configurations for resistive processing units.
aihwkit_lightning.simulator.configs.configs	Configurations for resistive processing units.
aihwkit_lightning.simulator.parameters	RPU simulator bindings.
aihwkit_lightning.simulator.parameters.enums	Utility enumerators for resistive processing units.
aihwkit_lightning.simulator.parameters.helpers	Utilities for resistive processing units configurations.
aihwkit_lightning.simulator.parameters.inference	Inference related parameters for resistive processing units.
aihwkit_lightning.simulator.parameters.io	Forward related parameters for resistive processing units.
aihwkit_lightning.simulator.parameters.mapping	Mapping parameters for resistive processing units.
aihwkit_lightning.simulator.parameters.pre_post	Pre-post processing related parameters for resistive processing units.