* Add mlx extra ops

* Add log_softmax dispatch for mlx

* Fix split bug, add tests

* Feedback