AI Model Optimization Specialist

Currently I work at a High Performance Computing centre in Germany, optimizing AI models, chiefly by writing custom CUDA/C++ PyTorch extensions. I am searching for more work along these lines, in either public or private institutions, ideally in an English speaking country.