Fixes JAX intro lecture

[kp992]

No description provided.

[netlify]

✅ Deploy Preview for incomparable-parfait-2417f8 ready!

Name	Link
🔨 Latest commit	e353bdb
🔍 Latest deploy log	https://app.netlify.com/projects/incomparable-parfait-2417f8/deploys/68b90f04b9660b0007341447
😎 Deploy Preview	https://deploy-preview-162--incomparable-parfait-2417f8.netlify.app
📱 Preview on mobile	Toggle QR Code... Use your smartphone camera to open QR code link.

---

To edit notification comments on pull requests, go to your Netlify project configuration.

[github-actions]

🚀 Deployed on https://68b91188c90c427c2dbf09e2--incomparable-parfait-2417f8.netlify.app

[mmcky]

thanks @kp992 -- lots of changes here. Can you please fill in the top level comment box with a description of the change etc. (cc @jstac)

[kp992]

Thanks @mmcky. This is just a draft PR. Will work on finishing it soon.

[mmcky]

@kp992 is this still in work?

[kp992]

I had just pasted @jstac suggestions shared over the email. @jstac Do you want me to finish this or are there changes that are obsolete?

[mmcky]

@kp992 could you please update this to the latest main and I will do a full review of the suggested changes.

[kp992]

Thanks for the reminder @mmcky . I will update the PR.