LEARNING HIERARCHICAL REPRESENTATIONS FOR EXPLAINABLE CHEMICAL REACTION PREDICTION