IBM Analog Hardware Acceleration Kit

Get started

  • Installation
  • Advanced installation guide
  • Using the PyTorch integration
  • Glossary

Analog AI Concepts

  • Analog AI
  • Analog AI Hardware
  • Advantages and Challenges

Cloud/Composer

  • Analog AI Cloud Composer Overview
  • Composer CLI

Using the Simulator

  • Using aihwkit Simulator

Analog DNN Training

  • Specialized Update Algorithms
  • Analog Training Presets

Analog DNN Inference

  • Inference and PCM Statistical Model
  • Analog Hardware-aware Training
  • Inference with Analog CMO-ReRAM Statistical Model

Advanced Guides

  • aihwkit design
  • Development setup
  • Development conventions
  • Project roadmap
  • Changelog

References

  • API Reference
  • Paper References
IBM Analog Hardware Acceleration Kit
  • aihwkit.utils package
  • View page source

aihwkit.utils package

Utilities and helpers for aihwkit.

Submodules

  • aihwkit.utils.analog_info module
    • AnalogInfo
      • AnalogInfo.calculate_num_analog()
      • AnalogInfo.calculate_num_tiles()
      • AnalogInfo.create_layer_summary()
      • AnalogInfo.register_hooks_recursively()
    • LayerInfo
      • LayerInfo.calculate_reuse_factor()
      • LayerInfo.input_size
      • LayerInfo.isanalog
      • LayerInfo.kernel_size
      • LayerInfo.layer_summary_dict()
      • LayerInfo.module
      • LayerInfo.name
      • LayerInfo.num_tiles
      • LayerInfo.output_size
      • LayerInfo.reuse_factor
      • LayerInfo.set_kernel_size()
      • LayerInfo.set_tiles_info()
      • LayerInfo.tiles_info
      • LayerInfo.tiles_summary_dict()
    • TileInfo
      • TileInfo.log_in_size
      • TileInfo.log_out_size
      • TileInfo.phy_in_size
      • TileInfo.phy_out_size
      • TileInfo.tile_summary_dict()
      • TileInfo.utilization
    • 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()

© Copyright 2020, 2021, 2022, 2023, 2024 IBM Research.

Built with Sphinx using a theme provided by Read the Docs.