Coursera Plus Annual is 40% Off — Ends July 6.

Class Central is learner-supported. When you buy through links on our site, we may earn an affiliate commission.

YouTube

JAX and FLAX for Neural Networks - Transformers

Discover AI via YouTube

Start learning Write review
DataCamp Ad

PowerBI Data Analyst - Create visualizations and dashboards from scratch

Learn More → Boot.dev Ad

The Fastest Way to Become a Backend Developer Online

Learn More →

Overview

AI, Data Science & Cloud Certificates from Google, IBM & Meta — 40% Off
One plan covers every Professional Certificate on Coursera. 40% off Coursera Plus Annual.
Unlock All Certificates
Learn to build transformer models including large language models and code LLMs using JAX and FLAX through this comprehensive tutorial covering functional programming principles, framework comparisons with PyTorch, and advanced topics like PyTrees for optimal parallelization in cloud infrastructure deployments.

Syllabus

WHY JAX? Why the Hell a 3rd ML framework in 2023?
Why FLAX Could Be Your New Favorite Deep Learning Library for NN
Intro to KERAS 3 (KERAS core) for PyTorch & JAX
Introduction to JAX 2023
JAX compared to PyTorch 2: Get a feeling for JAX!
Your next ML (Cloud) Infrastructure for your Code
PyTrees: Optimal Data Structure for JAX Parallelization

Taught by

Discover AI

Reviews

Start your review of JAX and FLAX for Neural Networks - Transformers

Start learning

Never Stop Learning.

Get personalized course recommendations, track subjects and courses with reminders, and more.

Sign up for free
Someone learning on their laptop while sitting on the floor.