GitLab

* another fix

* another try

* fix

* rewriting with kernel functions

* revert mxnet softmax changes

* lint fix
Co-authored-by: Zihao Ye <zihaoye.cs@gmail.com>