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.experiments

Experiments for aihwkit.

aihwkit.simulator

RPU simulator bindings.

aihwkit.simulator.configs

Configurations for resistive processing units.

aihwkit.simulator.noise_models

Phenomenological noise models for inference.

aihwkit.simulator.presets

Configurations presets for resistive processing units.

aihwkit.simulator.tiles

High level analog tiles.

aihwkit.nn

Neural network modules.

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.optim

Analog Optimizers.

aihwkit.optim.analog_sgd

Analog-aware stochastic gradient descent optimizer.

aihwkit.utils

Utilities and helpers for aihwkit.

aihwkit.utils.visualization

Visualization utilities.