JAX in 100 Seconds

Try Brilliant free for 30 days https://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  

🔗 Resources

Google JAX on GitHub https://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 https://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 Learning