aihwkit package

Analog hardware library for PyTorch.

Subpackages

• aihwkit.cloud package
  • Subpackages
    • aihwkit.cloud.client package
      • Subpackages
      • Submodules
    • aihwkit.cloud.converter package
      • Subpackages
      • Submodules
• aihwkit.experiments package
  • Subpackages
    • aihwkit.experiments.experiments package
      • Submodules
    • aihwkit.experiments.runners package
      • Submodules
• aihwkit.extension package
  • Submodules
    • aihwkit.extension.functions module
• aihwkit.inference package
  • Subpackages
    • aihwkit.inference.calibration package
      • Submodules
    • aihwkit.inference.compensation package
      • Submodules
    • aihwkit.inference.converter package
      • Submodules
    • aihwkit.inference.noise package
      • Submodules
  • Submodules
    • aihwkit.inference.utils module
      • drift_analog_weights()
      • program_analog_weights()
• aihwkit.nn package
  • Subpackages
    • aihwkit.nn.modules package
      • Subpackages
      • Submodules
  • Submodules
    • aihwkit.nn.conversion module
      • convert_to_analog()
      • convert_to_analog_mapped()
      • convert_to_digital()
      • specific_rpu_config_id()
• aihwkit.optim package
  • Submodules
    • aihwkit.optim.analog_optimizer module
      • AnalogAdam
      • AnalogOptimizer
      • AnalogOptimizerMixin
      • AnalogSGD
    • aihwkit.optim.context module
      • AnalogContext
• aihwkit.simulator package
  • Subpackages
    • aihwkit.simulator.configs package
      • Submodules
    • aihwkit.simulator.parameters package
      • Submodules
    • aihwkit.simulator.presets package
      • Submodules
    • aihwkit.simulator.tiles package
      • Submodules
  • Submodules
    • aihwkit.simulator.noise_models module
    • aihwkit.simulator.rpu_base module
      • Submodules
• aihwkit.utils package
  • Submodules
    • aihwkit.utils.analog_info module
      • AnalogInfo
      • LayerInfo
      • TileInfo
      • analog_summary()
    • aihwkit.utils.export module
      • fusion_export()
      • fusion_import()
    • aihwkit.utils.fitting module
      • fit_measurements()
      • model_response()
    • aihwkit.utils.legacy module
      • convert_legacy_checkpoint()
    • aihwkit.utils.visualization module
      • compute_pulse_response()
      • compute_pulse_statistics()
      • estimate_n_steps()
      • get_tile_for_plotting()
      • plot_device()
      • plot_device_compact()
      • plot_device_symmetry()
      • plot_model_programming_error()
      • plot_programming_error()
      • plot_pulse_response()
      • plot_pulse_statistics()
      • plot_response_overview()
      • plot_weight_drift()
    • aihwkit.utils.visualization_web module
      • plot_device_compact_web()
      • save_plots_for_web()
      • set_dark_style()

Submodules

• aihwkit.exceptions module
  • AihwkitException
  • AnalogBiasConfigError
  • ArgumentError
  • CloudError
  • ConfigError
  • CudaError
  • FusionExportError
  • ModuleError
  • TileError
  • TileModuleError
  • TorchTileConfigError
• aihwkit.version module