Hey there!
Indeed you can be arrested for DUI under the influence of a stimulant.
While some think that you can be arrested for driving under influence only because of alcohol, you can be arrested for using any drug while driving, even stimulants, because they are still impacting you in a way that can change behavior and impair your ability to drive- endangering other drivers.
Hope this helps!