feat: add softmax operation support#17

Jah-yee wants to merge 2 commits into

thinking-machines-lab:mainthinking-machines-lab/batch_invariant_ops:mainfrom

Jah-yee:softmax-opJah-yee/batch_invariant_ops:softmax-opCopy head branch name to clipboard