Augmenting Quantum Mechanics with Artificial Intelligence