API ReferenceΒΆ

aihwkit

Analog hardware library for PyTorch.

aihwkit.cloud

Functionality related to the cloud client for AIHW Composer API.

aihwkit.cloud.client

Client for connecting to the the AIHW Composer API.

aihwkit.cloud.converter

Conversion utilities for interacting with the AIHW Composer API.

aihwkit.exceptions

Custom Exceptions for aihwkit.

aihwkit.experiments

High-level interface for executing Experiments.

aihwkit.experiments.experiments

Experiments for aihwkit.

aihwkit.experiments.runners

Experiment Runners for aihwkit.

aihwkit.inference

High level inference tools.

aihwkit.inference.compensation

Compensation methods such as drift compensation during analog inference.

aihwkit.inference.converter

Converter of weight matrix values into conductance values and back for analog inference.

aihwkit.inference.noise

Noise models to apply to converted weight values during analog inference.

aihwkit.simulator

RPU simulator bindings.

aihwkit.simulator.configs

Configurations for resistive processing units.

aihwkit.simulator.presets

Configurations presets for resistive processing units.

aihwkit.simulator.tiles

High level analog tiles.

aihwkit.nn

Neural network modules.

aihwkit.nn.conversion

Digital/analog model conversion utilities.

aihwkit.nn.functions

Autograd functions for aihwkit.

aihwkit.nn.modules

Neural network modules.

aihwkit.nn.modules.base

Base class for analog Modules.

aihwkit.nn.modules.container

Analog Modules that contain children Modules.

aihwkit.nn.modules.conv

Convolution layers.

aihwkit.nn.modules.linear

Analog layers.

aihwkit.nn.modules.lstm

Analog LSTM layers.

aihwkit.optim

Analog Optimizers.

aihwkit.optim.analog_optimizer

Analog-aware inference optimizer.

aihwkit.optim.context

Parameter context for analog tiles.

aihwkit.utils

Utilities and helpers for aihwkit.

aihwkit.utils.visualization

Visualization utilities.

aihwkit.version

Package version string.