API Reference

aihwkit	Analog hardware library for PyTorch.
aihwkit.simulator	RPU simulator bindings.
aihwkit.simulator.devices	High level analog devices.
aihwkit.simulator.parameters	Parameters for resistive devices and tiles.
aihwkit.simulator.rpu_base
aihwkit.simulator.tiles	High level analog tiles.
aihwkit.simulator.tiles_indexed	High level analog tiles (indexed).
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.