A Geometric Framework for Neural Feature Learning