Add axes argument to Transpose Rust Kernel.

[yanndupis]

In the Python kernel, we support axes argument (see numpy for examples) but not in the Rust kernel.

[yanndupis]

It's still hasn't been implemented. It's not required by any model right now. But it's a feature supported by numpy, pytorch etc.

[yanndupis]

Also it's an optional arg on in the edsl but not used by the rust kernel. So we should complete the rust kernel or remove the arg in the eds.