346

Introduction to CUDA programming for Python developers

Stupid question: Is there any chance that I, as an engineer, can get away from learning the Math side of AI but still drill deeper into the lower level of CUDA or even GPU architecture? If so, how do I start? I guess I should learn about optimization and why we chose to use GPU for certain computations.

Parallel question: I work as a Data Engineer and always wonder if it's possible to get into MLE or AI Data Engineering without knowing AI/ML. I thought I only need to know what the data looks like, but so far I see every job description of an MLE requires background in AI.

a day agoferguess_k

Yes. They are largely unrelated. Just go to Nvidia's site and find the docs. Or there are several books (look at amazon).

A "background in AI" is a bit silly in most cases these days. Everyone is basically talking about LLMs or multimodal models which in practice haven't been around long. Sebastian Raschka has a good book about building an LLM from scratch, Simon Prince has a good book on deep learning, Chip Huyen has a good book on "AI engineering". Make a few toys. There you have a "background".

Now if you want to really move the needle... get really strong at all of it, including PTX (nvidia gpu assembly, sort of). Then you can blow people away like the deep seek people did...

a day agodanielmarkbruce

Lets say you already have deep knowledge of GPU architecture and experience optimizing GPU code to saves 0.5ms runtime for a kernel. But you got that experience from writing graphics code for rendering, and have little knowledge of AI stuff beyond surface level stuff of how neural networks work.

How can I leverage that experience into earning the huge amounts of money that AI companies seem to be paying? Most job listings I've looked at require a PhD in specifically AI/math stuff and 15 years of experience (I have a masters in CS, and no where close to 15 years of experience).

a day agojms55

I've only done the CUDA side (and not professionally), so I've always wondered how much those skills transfer either way myself. I imagine some of the specific techniques employed are fairly different, but a lot of it is just your mental model for programming, which can be a bit of a shift if you're not used to it.

I'd think things like optimizing for occupancy/memory throughput, ensuring coalesced memory accesses, tuning block sizes, using fast math alternatives, writing parallel algorithms, working with profiling tools like nsight, and things like that are fairly transferable?

a day agosuresk

I don't have a great answer except learn as much about AI as possible - the easiest starting point is Simon Prince's book - and it's free online. Maybe start submitting changes to pytorch? Get a name for yourself? I don't know.

Most companies aren't doing a lot of heavy GPU optimization. That's why deepseek was able to come out of nowhere. Most (not all) AI research basically takes the given hardware (and most of the software) stack as a given and is about architecture, loss functions, data mix, activation functions blah blah blah.

Speculation - a good amount of work will go towards optimizations in future (and at the big shops like openAI, a good amount already is).

a day agodanielmarkbruce

Is this hypothetical person someone you know? if yes, please email me to pavel at centml dotz ai

a day agopavelstoev

You can get paid that without the GPU experience so yes. Getting up to speed with this is mostly just a function of how able you are to understand what modern ML architectures look like.

a day agosaagarjha

Thank you! This really helps. I'll concentrate on Computer Architecture and lower level optimization then. I'll also pick one of the books just to get some ideas.

a day agoferguess_k

Agreed, Rashka's book is amazing and will probably become the seminal book on LLMs

a day agot55

Just to add that he has a video series on DL (youtube), completely approachable and accompanied by code notebooks.

20 hours agobarrenko

How does it compare with Andrej Karpathy’s video series on building GPTs from scratch? Are they pretty much teaching the same things?

17 hours agora7

Karpathy focuses on GPT, well, NLP-related specifics, while Raschka overviews Deep Learning as a whole, starting from the Perceptron basically.

Karpathy's teaching style is well, Karpathy, Raschka is more conventional (but not buttoned down).

15 hours agobarrenko

The math isn't that difficult. The transformers paper (https://proceedings.neurips.cc/paper_files/paper/2017/file/3...) was remarkably readable for such a high impact paper. Beyond the AI/ML specific terminology (attention) that were thrown out

Neural networks are basically just linear algebra (i.e matrix multiplication) plus an activation function (ReLu, sigmoid, etc.) to generate non-linearities.

Thats first year undergrad in most engineering programs - a fair amount even took it in high school.

a day agoSJC_Hacker

I'd like to re-enforce this viewpoint. The math is non-trivial, but if you're a software engineer, you have the skills required to learn _enough_ of it to be useful in the domain. It's a subject which demands an enormous amount of rote learning - exactly the same as software engineering.

20 hours agoOtherShrezzing

hot take: i don't think you even need to understand much linear algebra/calculus to understand what a transformer does. like the math for that could probably be learned within a week of focused effort.

15 hours agot55

Yeah to be honest its mostly the matrix multiplication, which I got in second year algebra (high school)0.

You don't really need even need to know about determinants, inverting matrices, Gauss-Jordan elimination, eigenvalues, etc. that you'd get in a first year undergrad linear algebra

14 hours agoSJC_Hacker

Not a stupid question at all! Imo, you can definitely dive deep into CUDA and GPU architecture without needing to be a math whiz. Think of it like this: you can be a great car mechanic without being the engineer who designed the engine.

Start with understanding parallel computing concepts and how GPUs are structured for it. Optimization is key - learn about memory access patterns, thread management, and how to profile your code to find bottlenecks. There are tons of great resources online, and NVIDIA's own documentation is surprisingly good.

As for the data engineering side, tbh, it's tougher to get into MLE without ML knowledge. However, focusing on the data pipeline, feature engineering, and data quality aspects for ML projects might be

16 hours agocodelion

Thanks for the help!

> As for the data engineering side, tbh, it's tougher to get into MLE without ML knowledge. However, focusing on the data pipeline, feature engineering, and data quality aspects for ML projects might be

I have a feeling that companies usually expect MLE to do both ML/AI and Data Engineering, so this might indeed be a dead end. Somehow I'm just not very interested in the MLE part of ML so I'll dormant that thought for the meanwhile.

> Start with understanding parallel computing concepts and how GPUs are structured for it. Optimization is key - learn about memory access patterns, thread management, and how to profile your code to find bottlenecks. There are tons of great resources online, and NVIDIA's own documentation is surprisingly good.

Thanks a lot! I'll take these points in mind when learning. I need to go through more basic CompArch materials first I think. I'm not a good programmer :D

15 hours agoferguess_k

Agreed, not sure how much math is really needed.

15 hours agot55

I suggest having a look at https://m.youtube.com/@GPUMODE

They have excellent resources to get you started with Cuda/Triton on top of torch. It also has a good community around it so you get to listen to some amazing people :)

14 hours agomusebox35

It's definitely possible to focus on the CUDA/GPU side without diving deep into the math. Understanding parallel computing principles and memory optimization is key. I've found that focusing on specific use cases, like optimizing inference, can be a good way to learn. On that note, you might find https://github.com/codelion/optillm useful – it optimizes LLM inference and could give you practical experience with GPU utilization. What kind of AI applications are you most interested in optimizing?

a day agocodelion

> Math side of AI but still drill deeper into the lower level of CUDA or even GPU architecture

CUDA requires clear understanding of mathematics related to graphics processing and algebra. Using CUDA like you would use traditional CPU would yield abysmal performance.

> MLE or AI Data Engineering without knowing AI/ML

It's impossible to do so, considering that you need to know exactly how the data is used in the models. At the very least you need to understand the basics of the systems that use your data.

Like 90% of the time spent in creating ML based applications is preparing the data to be useful for a particular use case. And if you take Google ML Crash Course, you'll understand why you need to know what and why.

14 hours agoJAlexoid

I will provide general advice that applies here, and elsewhere: Start with a project, and implement it, using CUDA. The key will be identifying a problem that is SIMD in nature. Choose something you would normally use a loop for, but that has many (e.g. tens of thousands or more) iterations, which do not depend on the output of the other iterations.

Some basic areas to focus on:

  - Setting up the architecture and config
  - Learning how to write the kernels, and what makes sense for a kernel
  - Learning how the IO and synchronization between CPU and GPU work.
This will be as learning any new programming skill.
18 hours agothe__alchemist

IMO absolutely yes. I would start with the linked introduction and then ask myself if I enjoyed it.

for a deeper dive, check out the sth like Georgia Tech’s CS 8803 O21: GPU Hardware and Software.

To get into MLE/AI Data Engineering, I would start with a brief introductory ML course like Andrew Ng’s on Coursera

a day agot55

Thanks! I'll follow the link and see what happens. And thanks for recommending Andrew Ng's course too, hopefully it gives enough background to know how the users (AI scientists) want us to prepare the data.

a day agoferguess_k

If you want to dive into CUDA specifically then I recommend following some of the graphics tutorials. Then mess around with it yourself, trying to implement any cool graphic/visualization ideas or remixes on the tutorial material.

You could also try to recreate or modify a shader you like from https://www.shadertoy.com/playlist/featured

You'll inevitably pick up some of the math along the way and probably have fun doing it along the way.

a day agoFalimonda

Yes, but the problems that need GPU programming also tend to require you to have some understanding of maths. Not exclusively - but it needs to be a problem that's divisible into many small pieces that can be recombined at the end, and you need to have enough data to work through that the compute cost + data transfer cost is much lower than just doing it on CPU.

21 hours agophysicsguy

From an infrastructure perspective, If you have access to the hardware, a fun starting point is running NCCL tests across the infrastructure. Start with a single GPU, then 8 GPUs on a host, then 24 GPU multi hosts over IB or RoCE. You will get a feel for MPI and plenty of knobs to turn on the Kubernetes side.

19 hours agomoondev

I mean yes, but without knowing the maths then knowing how to optimize the maths is a bit useless?

At the very least you should know enough linear algebra that you understand scalar, vector and matrix operations against each of the others. You don't need to be able to derive back prop from first principles, but you should know what happens when you multiply a matrix by a vector and apply a non-linear function to the result.

a day agollm_trw

Thanks! Yeah I do know some Math. I'm not sure how much I need to know. I guess the more the merrier, but it would be nice to know a line that I don't need to cross to properly do my job.

a day agoferguess_k

It's a tough one, I've never seen a book that actually covers the _bare_ minimum of the maths you need for ML.

The little learner comes close but I'd only really suggest that to people who already know the maths because the presentation is very non-standard and can get very misleading.

If you're interested drop me a line on my profile email and I'll have a look at some numerical algebra books and papers to see what's out there.

a day agollm_trw

I found the gpumode lectures, videos and code right on the money. check them out.

16 hours agobwfan123

You will probably have fewer job opportunities than the people working higher up, but be safer from AI automation for now :)

a day agoesafak

Try dipping your toes into graphics programming, you can still use GPUs for that as well.

a day agofulafel

Very nice-write up. The in-line quiz, which i think is AI generated(QnA) is very useful to test understanding. Wish all tutorials incorporated that feature.

13 hours agoultrasounder

thank you!

12 hours agot55

Thanks for sharing, enjoyed reading it!

I have a slightly tangential question: Do you have any insights into what exactly DeepSeek did by bypassing CUDA that made their run more efficient?

I always found it surprising that a core library like Cuda, developed over such a long time, still had room for improvement—especially to the extent that a seemingly new team of developers could bridge the gap on their own.

a day agospps11

They didn’t. They used PTX, which is what CUDA C++ compiles down to, but which is part of the CUDA toolchain. All major players have needed to do this because the intrinsics for the latest accelerators are not actually exposed in the C++ API, which means using them requires inline PTX at the very minimum.

a day agosaagarjha

They basically ditched CUDA and went straight to writing in PTX, which is like GPU assembly, letting them repurposing some cores for communication to squeeze out extra performance. I believe that with better AI models and tools like Cursor, we will move to a world where you can mold code ever more specific to your use case to make it more performant.

a day agot55

Are you sure they ditched CUDA? I keep hearing this, but it seems odd because that would be a ton of extra work to entirely ditch it vs selectively employing some ptx in CUDA kernels which is fairly straightforward.

Their paper [1] only mentions using PTX in a few areas to optimize data transfer operations so they don't blow up the L2 cache. This makes intuitive sense to me, since the main limitation of the H800 vs H100 is reduced nvlink bandwidth, which would necessitate doing stuff like this that may not be a common thing for others who have access to H100s.

1. https://arxiv.org/abs/2412.19437

a day agosuresk

I should have been more precise, sorry. Didn't want to imply they entirely ditched CUDA but basically circumvented it in a few areas like you said.

15 hours agot55

Targeting directly PTX is perfectly regular CUDA, and used by many toolchains that target the ecosystem.

CUDA is not only C++, as many mistake it for.

14 hours agopjmlp

got it, thanks for explaining.

> with better AI models and tools like Cursor, we will move to a world where you can mold code ever more specific to your use case to make it more performant

what do you think the value of having the right abstraction will be in such a world?

a day agospps11

I think that for at least for us dumb humans with limited memory, having good abstractions makes things much easier to understand

a day agot55

Yes, but I wonder how much of this trait is carried over to the LLMs from us.

a day agospps11

what do you mean, the LLM abstracting things for us while we speak to it?

a day agot55

No I meant something else. As you said: us humans love clean abstractions. We love building on top of them. Now LLMs are trained on data produced by us. So I wonder if they would also inherit this trait from us and end up loving good abstractions, and would find it easier to build on top of them. Other possibility is that they end up move-37ing the whole abstraction shebang. And find that always building something up bespoke, from low-level is better than constraining oneself to some general purpose abstraction.

a day agospps11

ah gotcha. I think that with the new trend of RLing models, the move 37 may come up sooner than we think -- just provide the pretrained models some outcome-goal and the way it gets there may use low-level code without clean abstractions

a day agot55

It's an interesting idea.

If code is ever updated by an LLM, does it benefit from using abstractions? After all they're really a tool for us lowly sapients to aid in breaking down complex problems. Maybe LLM's will create their own class of abstractions, diverse from our own but useful for their task.

a day agotomnipotent

What Jensen giveth, Guido taketh away.

a day agomusicale

lol. i guess this tutorial is about cutting out guido ;)

a day agot55

this book:

    Programming Massively Parallel Processors by Wen-mei W. Hwu , David B. Kirk , Izzat El Hajj
seems to be tailor mode for folks transitioning from cpu -> gpu arch.
a day agosigna11

Yes, it is great for key concepts but a bit outdated. Hence we added an LLM/FA section in the linked post!

15 hours agot55

Are all the CUDA tutorials geared towards AI or are there some, for example, like regular scientific computing? Airflow over wings and things that you used to see for high-performance computing would be fun to try.

10 hours agoralphc

Related: https://sakana.ai/ai-cuda-engineer/

https://www.reddit.com/r/MachineLearning/comments/1itqrgl/p_...

a day agot55

Wasn’t this a bunch of kernels that didn’t work?

a day agosaagarjha

What do you mean?

a day agot55

The hallucinated code was reusing memory buffers filled with previous results so not performing the actual computations. When this was fixed the AI generated code was like 0.3x of the baseline.

a day agopavelstoev

It is mentioned on section "Limitations and Bloopers" of the page [0]:

> Combining evolutionary optimization with LLMs is powerful but can also find ways to trick the verification sandbox. We are fortunate to have Twitter user @main_horse help test our CUDA kernels, to identify that The AI CUDA Engineer had found a way to “cheat”. The system had found a memory exploit in the evaluation code which, in a small percentage of cases, allowed it to avoid checking for correctness (...)

0. https://sakana.ai/ai-cuda-engineer

a day agoneodypsis

As I write this (after the updates to the evaluation code), https://pub.sakana.ai/ai-cuda-engineer/kernel/2/23/optimize-... is on their top of their list of speedups, with a claim of 128x speed up on a fused 3D convolution + groupnorm + mean.

The generated implementation doesn’t do a convolution.

The 2nd kernel on the leaderboard also appears to be incorrect, with a bunch of dead code computing a convolution and then not using it and writing tanhf(1.0f) * scaling_factor for every output.

a day agornrn

They don't verify the correctness of their kernels. They expect you to pick the working ones from their kernel junkyard yourself.

The very idea is also dumb as hell. They could have done CUDA -> HIP/oneAPI/Metal/Vulkan/SYCL/OpenCL. Then they wouldn't need to beat the performance of anything, just the automatic porting would be worth an acquisition by AMD or Intel.

a day agoimtringued

Problem with startups like Devin (AI sw engineer) and Sakana (AI research scientist) is that they are full of hot-air.

They get caught up in the hype, and focus on the marketing and not the essential engineering.

16 hours agobwfan123

[dead]

a day agotsunego

Since this is on PySpur's website, does anyone have experience with these UI tools for AI agents like PySpur and n8n? I am looking for something to help me prototype a few ideas for fun. I would have to self-host it ($), so I would prefer something relatively easy to configure like Open Hands.

a day agom_kos

Disclaimer: I work on pyspur

I'd recommend pyspur if you seek

1) More AI-native features eg. Evals, RAG, or even UI decisions like seeing outputs directly on the canvas when running on the agent 2) Truly open-source Apache license 3) Python-based (in the sense that you can run and extend it via python)

On the other hand, n8n is 1) more mature for traditional workflows 2) offering overall more integrations (probably every single integration you can think of) 3) TypeScript based and runs on Node.js

a day agot55

Thanks for replying. Do you know when your docs will be a bit more comprehensive? Right now, there is very little information and some links don't work, e.g., Next Steps on this page: https://docs.pyspur.dev/quickstart

a day agom_kos

> Do you know when your docs will be a bit more comprehensive?

Yes, we're actively working on this, and we should have some more pages by next week. If you have any questions, you can always shoot us an email: founders@pyspur.dev or join our Discord.

> some links don't work, e.g., Next Steps on this page

This might be confusing, the cards below "After installation, you can:" are not meant to be links. Thanks for making us aware, we will improve the wording.

a day agot55

pyspur is apache 2. it is free to self-host.

a day agospps11

Any idea what changed recently and we can have end to end simulations (with branches) in the gpu (eg isaac sim) vs in the past where simulations were a cpu thing ?

20 hours agowhatever1

Always been possible, but now the time cost of moving data between the GPU and CPU memory is too high to ignore. Branching may be slower on the GPU but it's still faster than moving data to the CPU for a time then back. The maturation of direct GPU-GPU transfers over the network also helped enable GPU-only MPI codes.

20 hours agojamiejquinn

Interestingly, the CUDA implementations are more readable than the pytorch ones.

a day agortkal10

interesting, you mean they are less obscure?

a day agot55

[dead]

a day agotsunego

If you are a Python dev, why not just use Triton?

a day agonitrogen99

Triton sits between CUDA and PyTorch and is built to work smoothly within the PyTorch ecosystem. In CUDA, on the other hand, you can directly manipulate warp-level primitives and fine-tune memory prefetching to reduce latency in eg. attention algorithms, a level of control that Triton and PyTorch don't offer AFAIK.

a day agot55

MLIR extensions for Python do though, as far as I could tell from LLVM developer meeting.

14 hours agopjmlp

MLIR is one of those things everyone seems to use, but nobody seems to want to write solid introductory docs for :(

I've been curious for a few years now to get into MLIR, but I don't know compilers or LLVM, and all the docs I've found seem to assume knowledge of one or the other.

(yes this is a plea for someone to write an 'intro to compilers' using MLIR)

14 hours ago6gvONxR4sf7o

Triton is somewhat limited in what it supports, and it’s not really Python either.

a day agosaagarjha

or use Hidet compiler (open source)

a day agopavelstoev

never heard of Hidet before; for when/what would I use it over CUDA/Triton/Pytorch?

a day agot55

It is written in Python itself and emits efficient CUDA code. This way, you can understand what is going on. The current focus is on inference, but hopefully, training workloads will be supported soon. https://github.com/hidet-org/hidet

11 hours agopavelstoev

Thanks for unraveling this!

16 hours agoAchintyaAshok

you're welcome!

15 hours agot55

pyspur graph is cool, is there a startup building this kind of product but in typescript?

20 hours agoandroid521

I needed this

a day agolukaspetersson

Hehe glad you did!

a day agot55

[dead]