AttributeError: 'module' object has no attribute 'sum_axis'

Notebooks data.ipynb and matrix_factorization.ipynb are populated with
```python
mx.sym.sum_axis()
```
which raises the attribute error.

Guess should be
```python
mx.sym.sum()
```