Which library to choose for GPU computing python3 linux?

Question

linux

I wrote a program that implements self-learning of a population using a genetic algorithm in python3 + pygame. All calculations are performed on the CPU.
So far, the CPU performance is enough, but I want to write a more complex program with more calculations. The CPU from such a wave of calculations will simply choke! Help with information about GPU computing, I have never done this. I have been asking this question for quite some time, but most of the articles just show primitive examples, and nothing really explains. And without systematized knowledge, nothing happens. So far, the only thing I've figured out is that you need to use NumPy arrays to store data.
I will be glad to any systematized information! From books to articles and videos in Russian and English. Preferably in Russian. )