Talk: Learning JAX for Great Good

Image credit: Google JAX

Abstract

In this tutorial I introduce JAX, the new google framework for deep learning. We first introduce the main peculiarities and describe how to compute gradients to optimize simple ML algorithms. Eventually, we implement more complex NLP pipelines, showing were JAX modularity comes to shine.

Date
Feb 17, 2021 1:00 PM
Event
Invited Talk
Location
Coveo Labs