API Reference

aihwkit_lightning

Analog hardware library for PyTorch.

aihwkit_lightning.exceptions

Custom Exceptions for aihwkit.

aihwkit_lightning.version

Package version string.

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.