aihwkit.nn package

Neural network modules.

Subpackages

• aihwkit.nn.modules package
  • Subpackages
    • aihwkit.nn.modules.rnn package
      • Submodules
  • Submodules
    • aihwkit.nn.modules.base module
    • aihwkit.nn.modules.container module
    • aihwkit.nn.modules.conv module
    • aihwkit.nn.modules.conv_mapped module
    • aihwkit.nn.modules.linear module
    • aihwkit.nn.modules.linear_mapped module

Submodules

• aihwkit.nn.conversion module
• aihwkit.nn.functions module