Quite a few years ago I worked on a particle in cell simulation software during my bachelor thesis[0]. We used numba to accelerate the code and most importantly

Quite a few years ago I worked on a particle in cell simulation software during my bachelor thesis[0]. We used numba to accelerate the code and most importantly write GPU kernels for the heavy parts. I remember spending hours optimising my code to eek out the most performance possible (which eventually meant using atomics and manually unrolling many loops because somehow this was giving us the best performance) but honestly I was really happy that I didn't need to write cuda kernels in C and generally it was pretty easy to work with. I remember back then the documentation was sometimes a little rough around the edges but the numba team was incredibly helpful and responsive. Overall I had a great time. [0] https://github.com/fbpic/fbpic https://github.com/fbpic/fbpic