Understanding JAX - JIT, XLA, and Pure Functions Explained

Understanding JAX - JIT, XLA, and Pure Functions Explained

Google Developers via YouTube Direct link

Introduction to JAX concepts

1 of 9

1 of 9

Introduction to JAX concepts

Class Central Classrooms beta

YouTube videos curated by Class Central.

Classroom Contents

Understanding JAX - JIT, XLA, and Pure Functions Explained

Automatically move to the next video in the Classroom when playback concludes

  1. 1 Introduction to JAX concepts
  2. 2 What is JAX?
  3. 3 Functional Purity in JAX
  4. 4 Non-Modifiable Arrays
  5. 5 Explicit State Management
  6. 6 Pseudo Random Number Generation PRNG
  7. 7 Just-In-Time JIT Compilation
  8. 8 JIT Limitations & Control Flow
  9. 9 Summary & Conclusion

Never Stop Learning.

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

Someone learning on their laptop while sitting on the floor.