GradTree: Learning Axis-Aligned Decision Trees with Gradient Descent