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
https://discord.gg/fireship
🔗 Resources
Google JAX on GitHub https://github.com/google/jax
Tensorflow in 100 Seconds https://youtu.be/i8NETqtGHms
Cuda in 100 Seconds https://youtu.be/pPStdjuYzSI
GPU vs TPU https://youtu.be/r5NQecwZs1A
🔥 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