Concise Implementation of Softmax Regression

English Version jax
1

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