Add JAX NN training lecture

[jstac]

No description provided.

[netlify]

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

Name	Link
🔨 Latest commit	41982f9
🔍 Latest deploy log	https://app.netlify.com/sites/incomparable-parfait-2417f8/deploys/67d4a8f5f2312300097290e2
😎 Deploy Preview	https://deploy-preview-205--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 site configuration.

[github-actions]

🚀 Deployed on https://67d4ad8e0c54e62725011453--incomparable-parfait-2417f8.netlify.app

[jstac]

Hi @mmcky, are we not running deployments in this lecture set?

If not that's fine but could you please cast your eyes over it, fix any minor issues and make it live? Thanks!

[jstac]

@mmcky Okay, please ignore that comment about deployments :-)

Once I reloaded the page I got the link.

[jstac]

Fixed a couple of small typos. Over to you @mmcky

[mmcky]

thanks @jstac.

Indeed. Builds and publishing pipeline is all working. I have a couple of open PRs looking at ways to improve our runtimes and update to the latest cuda=12.8 which is proving problematic (for some reason the driver on the docker container is not being recognised with the current github actions image of ubuntu)

[mmcky]

@jstac really like this idea of breaking down some of the underlying routines and exposing how they work. Nice lecture!

[jstac]

Thanks @mmcky , glad you like it. Please feel free to merge when ready.

Can you also please make it live? Thanks.