API ReferenceΒΆ

aihwkit

Analog hardware library for PyTorch.

aihwkit.simulator

RPU simulator bindings.

aihwkit.simulator.configs

Configurations for resistive processing units.

aihwkit.simulator.noise_models

Phenomenological noise models for inference.

aihwkit.simulator.rpu_base

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

Convolution layers.

aihwkit.nn.modules.linear

Analog layers.

aihwkit.optim

Analog Optimizers.

aihwkit.optim.analog_sgd

Analog-aware stochastic gradient descent optimizer.