GPU Programming in Pure Python

If you're looking to leverage the insane power of modern GPUs for data science and ML, you might think you'll need to use some low-level programming language such as C++. But the folks over at NVIDIA have been hard at work building Python SDKs which provide nearly native level of performance when doing Pythonic GPU programming. Bryce Adelstein Lelbach is here to tell us about programming your GPU in pure Python.

Episode sponsors

Posit
Agntcy
Talk Python Courses

Links from the show

Bryce Adelstein Lelbach on Twitter: @blelbach

Episode Deep Dive write up: talkpython.fm/blog

NVIDIA CUDA Python API: github.com
Numba (JIT Compiler for Python): numba.pydata.org
Applied Data Science Podcast: adspthepodcast.com
NVIDIA Accelerated Computing Hub: github.com
NVIDIA CUDA Python Math API Documentation: docs.nvidia.com
CUDA Cooperative Groups (CCCL): nvidia.github.io
Numba CUDA User Guide: nvidia.github.io
CUDA Python Core API: nvidia.github.io
Numba (JIT Compiler for Python): numba.pydata.org
NVIDIA’s First Desktop AI PC ($3,000): arstechnica.com
Google Colab: colab.research.google.com
Compiler Explorer (“Godbolt”): godbolt.org
CuPy: github.com
RAPIDS User Guide: docs.rapids.ai
Watch this episode on YouTube: youtube.com
Episode #509 deep-dive: talkpython.fm/509
Episode transcripts: talkpython.fm

--- Stay in touch with us ---
Subscribe to Talk Python on YouTube: youtube.com
Talk Python on Bluesky: @talkpython.fm at bsky.app
Talk Python on Mastodon: talkpython
Michael on Bluesky: @mkennedy.codes at bsky.app
Michael on Mastodon: mkennedy