JAX in 100 Seconds

Try Brilliant free for 30 days brilliant.org/fireship You’ll also get 20% off an annual premium subscription

JAX is a Python library similar to NumPy for scientific computing and linear algebra, but designed to run on accelerators like Cuda-based GPUs and Google's TPUs.

#programming #math #100SecondsOfCode

💬 Chat with Me on Discord

discord.gg/fireship

🔗 Resources

Google JAX on GitHub github.com/google/jax
Tensorflow in 100 Seconds    • TensorFlow in 100 Seconds  
Cuda in 100 Seconds    • Nvidia CUDA in 100 Seconds  
GPU vs TPU    • CPU vs GPU vs TPU vs DPU vs QPU  


🔥 Get More Content - Upgrade to PRO

Upgrade at fireship.io/pro
Use code YT25 for 25% off PRO access

🎨 My Editor Settings

Atom One Dark
vscode-icons
Fira Code Font

🔖 Topics Covered

JAX basics tutorial
How does JAX work?
JAXPpython explained quickly
Pytorch alternative
Tensorflow vs JAX
How is calculus used on Machine Lea