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
Corporate Finance Institute Ad

Master Finance Tools - 35% Off CFI (Code CFI35)

Learn More → Coursera Ad

Save 40% on 3 months of Coursera Plus

Learn More →

Overview

Coursera Flash Sale
40% Off Coursera Plus for 3 Months!
Grab it
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.