Machine Learning with JAX - From Zero to Hero | Tutorial #1

❤️ Become The AI Epiphany Patreon ❤️ 👨‍👩‍👧‍👦 Join our Discord community 👨‍👩‍👧‍👦 With this video I’m kicking off a series of tutorials on JAX! JAX is a powerful and increasingly more popular ML library built by the Google Research team. The 2 most popular deep learning frameworks built on top of JAX are Haiku (DeepMInd) and Flax (Google Research). In this video I cover the basics as well as the nitty-gritty details of jit, grad, vmap, and various other idiosyncrasies of JAX. ▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬ ✅ JAX GitHub: ✅ JAX docs: ✅ My notebook: ✅ Useful video on autodiff: ▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬▬ ⌚️ Timetable: 00:00:00 What is JAX? JAX ecosystem 00:03:35 JAX basics 00:10:05 JAX is accelerator agnostic 00:15:00 jit explained 00:17:45 grad explained 00:27:25 The power of JAX autodiff (He