Amazon (amazon.com side)'s revenue depends on consumer behavior. If marijuana use in the general population is thought to change consumer behavior in their favor, then they would be foolish not to support it. This effect probably outweighs any worker productivity issues that might be introduced.

Use has been correlated with increased impulsive behavior: https://www.ncbi.nlm.nih.gov/pmc/articles/PMC4330120/