2D materials for logic and memory integration