Skip to content

Instantly share code, notes, and snippets.

View hzarei4's full-sized avatar

Hossein Zarei Oshtolagh hzarei4

  • Jena, Germany
View GitHub Profile
@vankesteren
vankesteren / ad_linreg.jl
Last active September 24, 2022 13:22
Gradient descent for linear regression using Zygote AutoDiff
# Gradient descent with autodiff for linear regression
using Zygote
# Data
X = randn(1000, 10)
b = (1:10)
y = X * b + randn(1000)
# MSE for linear model
function mse(bhat)