Gradient Descent with JAX

Code Labs Academy - Coding Bootcamp (

Date Time: February 11, 2023 | 1:00 pm
Time Zone: GMT+8:00 China Taiwan Time

Total Hours

1.5 Hrs

What is Google JAX? It’s a machine learning framework used for transforming numeral functions and is bringing together a modified version of autograd and TensorFlow’s XLA.


| Prerequisites ✅

Fundamentals in Python and Machine Learning are required. Check out “Introduction to JAX” before attending this workshop so you can get the most out of it.


| Start your journey in Tech 📅

Data Science Bootcamp

Gain the skills you need to make sense of the wealth of data available in our world. Data manipulation with Python and SQL; creation of machine learning algorithms for vision and natural language tasks, among other things!


Make sure to follow our Eventbrite profile for future workshops in Cybersecurity, Data Science, Web Development, and UX/UI Design!

Check out our social media below:





Please, keep in mind that these workshops are being recorded and shared on CLA’s Youtube channel for learning purposes. We want to make our videos easily accessible for everyone interested in the topic. Feel free to participate with your cameras on or off, whatever makes you feel more comfortable.


About the Event

In this workshop we will use Google JAX to implement gradient descent in order to solve machine learning tasks like linear regression and logistic regression. Gradient descent is the core of the training of all deep learning models. Implementing it from scratch using JAX will help you understand how this optimization algorithm works.