![](https://ruslar.me/mobile/static/img/info.png)
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
🔗 Resources
Google JAX on GitHub
Tensorflow in 100 Seconds
Cuda in 100 Seconds
GPU vs TPU
🔥 Get More Content - Upgrade to PRO
Upgrade at
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