aihwkit.simulator.configs package
Configurations for resistive processing units.
Submodules
- aihwkit.simulator.configs.compounds module
BufferedTransferCompoundChoppedTransferCompoundChoppedTransferCompound.as_bindings()ChoppedTransferCompound.auto_granularityChoppedTransferCompound.auto_momentumChoppedTransferCompound.auto_scaleChoppedTransferCompound.bindings_classChoppedTransferCompound.buffer_granularityChoppedTransferCompound.correct_gradient_magnitudesChoppedTransferCompound.fast_lrChoppedTransferCompound.forget_bufferChoppedTransferCompound.in_chop_probChoppedTransferCompound.in_chop_randomChoppedTransferCompound.momentumChoppedTransferCompound.no_bufferChoppedTransferCompound.out_chop_probChoppedTransferCompound.scale_transfer_lrChoppedTransferCompound.stepChoppedTransferCompound.transfer_columnsChoppedTransferCompound.transfer_forwardChoppedTransferCompound.transfer_lrChoppedTransferCompound.transfer_updateChoppedTransferCompound.units_in_mbatch
DifferenceUnitCellDigitalRankUpdateCellDynamicTransferCompoundDynamicTransferCompound.as_bindings()DynamicTransferCompound.bindings_classDynamicTransferCompound.buffer_capDynamicTransferCompound.buffer_granularityDynamicTransferCompound.experimental_correct_accumulationDynamicTransferCompound.fast_lrDynamicTransferCompound.forget_bufferDynamicTransferCompound.in_chop_probDynamicTransferCompound.in_chop_randomDynamicTransferCompound.momentumDynamicTransferCompound.scale_transfer_lrDynamicTransferCompound.stepDynamicTransferCompound.tail_weighteningDynamicTransferCompound.transfer_columnsDynamicTransferCompound.transfer_forwardDynamicTransferCompound.transfer_lrDynamicTransferCompound.transfer_update
MixedPrecisionCompoundMixedPrecisionCompound.as_bindings()MixedPrecisionCompound.bindings_classMixedPrecisionCompound.granularityMixedPrecisionCompound.n_d_binsMixedPrecisionCompound.n_rows_per_transferMixedPrecisionCompound.n_x_binsMixedPrecisionCompound.random_rowMixedPrecisionCompound.stoc_round_dMixedPrecisionCompound.stoc_round_xMixedPrecisionCompound.transfer_everyMixedPrecisionCompound.transfer_lr
OneSidedUnitCellReferenceUnitCellTransferCompoundTransferCompound.as_bindings()TransferCompound.bindings_classTransferCompound.fast_lrTransferCompound.gammaTransferCompound.gamma_vecTransferCompound.n_reads_per_transferTransferCompound.no_self_transferTransferCompound.random_selectionTransferCompound.scale_transfer_lrTransferCompound.transfer_columnsTransferCompound.transfer_everyTransferCompound.transfer_every_vecTransferCompound.transfer_forwardTransferCompound.transfer_lrTransferCompound.transfer_lr_vecTransferCompound.transfer_updateTransferCompound.units_in_mbatchTransferCompound.with_reset_prob
UnitCellVectorUnitCell
- aihwkit.simulator.configs.configs module
DigitalRankUpdateRPUConfigFloatingPointRPUConfigIOManagedRPUConfigInferenceRPUConfigInferenceRPUConfig.backwardInferenceRPUConfig.clipInferenceRPUConfig.compatible_with()InferenceRPUConfig.deviceInferenceRPUConfig.drift_compensationInferenceRPUConfig.forwardInferenceRPUConfig.modifierInferenceRPUConfig.noise_modelInferenceRPUConfig.remapInferenceRPUConfig.tile_array_classInferenceRPUConfig.tile_classInferenceRPUConfig.update
QuantizedTorchInferenceRPUConfigSingleRPUConfigTorchInferenceRPUConfigTorchInferenceRPUConfigIRDropTUnitCellRPUConfig
- aihwkit.simulator.configs.devices module
ConstantStepDeviceExpStepDeviceFloatingPointDeviceIdealDeviceLinearStepDeviceLinearStepDevice.allow_increasingLinearStepDevice.apply_write_noise_on_setLinearStepDevice.bindings_classLinearStepDevice.gamma_downLinearStepDevice.gamma_down_dtodLinearStepDevice.gamma_upLinearStepDevice.gamma_up_dtodLinearStepDevice.mean_bound_referenceLinearStepDevice.mult_noiseLinearStepDevice.reverse_downLinearStepDevice.reverse_offsetLinearStepDevice.reverse_upLinearStepDevice.write_noise_std
PiecewiseStepDevicePowStepDevicePowStepReferenceDevicePowStepReferenceDevice.bindings_classPowStepReferenceDevice.n_estimation_stepsPowStepReferenceDevice.pow_gammaPowStepReferenceDevice.pow_gamma_dtodPowStepReferenceDevice.pow_up_downPowStepReferenceDevice.pow_up_down_dtodPowStepReferenceDevice.reference_meanPowStepReferenceDevice.reference_stdPowStepReferenceDevice.subtract_symmetry_point
PulsedDevicePulsedDevice.as_bindings()PulsedDevice.bindings_classPulsedDevice.bindings_modulePulsedDevice.construction_seedPulsedDevice.corrupt_devices_probPulsedDevice.corrupt_devices_rangePulsedDevice.count_pulsesPulsedDevice.diffusionPulsedDevice.diffusion_dtodPulsedDevice.driftPulsedDevice.dw_minPulsedDevice.dw_min_dtodPulsedDevice.dw_min_dtod_log_normalPulsedDevice.dw_min_stdPulsedDevice.enforce_consistencyPulsedDevice.lifetimePulsedDevice.lifetime_dtodPulsedDevice.perfect_biasPulsedDevice.requires_decay()PulsedDevice.requires_diffusion()PulsedDevice.resetPulsedDevice.reset_dtodPulsedDevice.reset_stdPulsedDevice.up_downPulsedDevice.up_down_dtodPulsedDevice.w_maxPulsedDevice.w_max_dtodPulsedDevice.w_minPulsedDevice.w_min_dtod
SoftBoundsDeviceSoftBoundsPmaxDeviceSoftBoundsReferenceDeviceSoftBoundsReferenceDevice.apply_write_noise_on_setSoftBoundsReferenceDevice.bindings_classSoftBoundsReferenceDevice.mult_noiseSoftBoundsReferenceDevice.reference_meanSoftBoundsReferenceDevice.reference_stdSoftBoundsReferenceDevice.slope_down_dtodSoftBoundsReferenceDevice.slope_up_dtodSoftBoundsReferenceDevice.subtract_symmetry_pointSoftBoundsReferenceDevice.write_noise_std
- aihwkit.simulator.configs.helpers module
- aihwkit.simulator.configs.utils module