Softmax Regression Implementation from Scratch

English Version jax
1

https://d2l.ai/chapter_linear-classification/softmax-regression-scratch.html