https://derekzhouai.github.io/posts/linear-regression-implementation-concise/