Power management based on detected voltage parameter levels in a memory sub-system

A non-transitory computer readable medium comprising instructions, which when executed by a processing device, cause the processing device to perform operations comprising:

• executing, in a training phase, a set of memory management operations on a memory die of a plurality of memory dies of a memory sub-system;
• identifying a peak current draw event produced by executing a memory management operation portion of the set of memory management operations;
• detecting a voltage droop level corresponding to the peak current draw event;
• and
• storing a token representing the voltage droop level and information identifying the memory management operation portion.

A system comprising:

• a memory device;
• and
• a processing device, operatively coupled with the memory device, to perform operations comprising:
• initiating a command for execution of a set of memory management operations by a plurality of memory dies of a memory sub-system;
• executing the set of memory management operations on each memory die of the plurality of memory dies;
• determining a set of voltage parameter levels corresponding to execution of the set of memory management operations;
• determining, based on the set of voltage parameter levels, information representing a voltage parameter level and a corresponding portion of the set of memory management operations;
• receiving a request from a host system to execute a target portion of a memory management operation;
• identifying first information corresponding to the target portion of the memory management operation;
• and
• performing a power management action based on a first voltage parameter level of the first information.